unflatten

Contents

unflatten#

class brainstate.graph.unflatten(graph_def, state_mapping, /, *, index_ref=None, index_ref_cache=None)#

Unflatten a graph_def + state_mapping back into a node.

Return type:

Any