tfp.experimental.mcmc.StateWithHistory

StateWithHistory(state, state_history)

state

state_history