Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tfp.experimental.substrates.numpy.mcmc.StatesAndTrace

View source on GitHub

States and auxiliary trace of an MCMC chain.

@staticmethod
tfp.experimental.substrates.numpy.mcmc.StatesAndTrace(
    _cls, all_states, trace
)

The first dimension of all the Tensors in this structure 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.