Integer number of results to expect (as passed to sample
chain).
progress_bar_fn
A function that takes an integer num_results and
returns an iterator that advances a progress bar. Defaults to tqdm
progress bars (make sure they are pip installed befure using.)
Finalizes target statistic calculation from the final_state.
This is an identity function of the final_state by default. Subclasses
can override it for streaming calculations whose running state is not the
same as the desired result.
Args
final_reducer_state
A tuple, namedtuple or list of Tensors
representing the final state of the reduced statistic.
A (possibly nested) structure of Tensors or Python
lists of Tensors representing the current state(s) of the Markov
chain(s). It is used to infer the structure of future trace results.
initial_kernel_results
A (possibly nested) structure of Tensors
representing internal calculations made in a related TransitionKernel.
It is used to infer the structure of future trace results.