treefy_states

Contents

treefy_states#

class brainstate.graph.treefy_states(node, *filters)#

Return the treefy state mapping of a graph node, optionally filtered.

Return type:

NestedDict | tuple[NestedDict, ...]