flatten

Contents

flatten#

class brainstate.graph.flatten(node, /, ref_index=None, treefy_state=True)#

Flatten a graph node into a (graph_def, state_mapping) pair.

Return type:

tuple[GraphDef[Any], NestedDict]