brainstate.graph.split_context

Contents

brainstate.graph.split_context#

brainstate.graph.split_context()[source]#

Context manager for splitting multiple graph nodes sharing a reference index.

Return type:

Iterator[Tuple[SplitContext, RefMap[Any, int]]]