EpochChainManager.combine_filtered()#

EpochChainManager.combine_filtered(predicate)[source]#

Combines the all epochs for which the predicate evaluates to True. Returns all chunks combined into one pytree.

The option is none if no samples are in the chain.

Return type:

Option[TypeVar(TPyTree, bound= Any)]