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