Registration is open for TensorFlow Dev Summit 2020 Learn more

tfp.experimental.substrates.jax.mcmc.CheckpointableStatesAndTrace

View source on GitHub

Class CheckpointableStatesAndTrace

States and auxiliary trace of an MCMC chain.

The first dimension of all the Tensors in the all_states and trace attributes is the same and represents the chain length.

Attributes:

  • all_states: A Tensor or a nested collection of Tensors representing the MCMC chain state.
  • trace: A Tensor or a nested collection of Tensors representing the auxiliary values traced alongside the chain.
  • final_kernel_results: A Tensor or a nested collection of Tensors representing the final value of the auxiliary state of the TransitionKernel that generated this chain.

__new__

@staticmethod
__new__(
    _cls,
    all_states,
    trace,
    final_kernel_results
)

Create new instance of CheckpointableStatesAndTrace(all_states, trace, final_kernel_results)

Properties

all_states

trace

final_kernel_results