class_docstr: Extra outputs for Adaptive HMC `TransitionOperator`.
docstr: Returns the chain state.
  
  Note that this assumes that `target_log_prob_fn` has the `state_extra` be a
  tuple, with the first element thereof containing the state in the original
  space.
docstr_tok: ["Returns","the","chain","state",".","Note","that","this","assumes","that","`","target_log_prob_fn","`","has","the","`","state_extra","`","be","a","tuple",",","with","the","first","element","thereof","containing","the","state","in","the","original","space","."]
code: def state(self) -> fun_mc.ArrayNest:
      """Returns the chain state.
  
      Note that this assumes that `target_log_prob_fn` has the `state_extra` be a
      tuple, with the first element thereof containing the state in the original
      space.
      """
      return self.hmc_state.state_extra[0]
  
imports: import functools
  import typing
  from typing import Any, Callable, Iterable, NamedTuple, Optional
  import numpy
  from fun_mc import backend
  from fun_mc import fun_mc_lib
  from fun_mc import malt
  from fun_mc import sga_hmc
  from fun_mc.malt import *
  from fun_mc.sga_hmc import *
label_desc: Use image node_id 1 for calling the AdaptiveHamiltonianMonteCarloExtra obj's underlying member method code with example usage: obj.state() and returns: self