API Reference
Returns all chunks combined into one pytree.
The option is none if no samples are in the chain.
Option[TypeVar(TPyTree, bound= Any)]
Option
TypeVar
TPyTree
Any