Traverse: Pytree traversal utility
Ops
Flatten an arbitrary python object into a tree definition and a collection of leaves. |
|
Repack a tree definition and list of leaves into the original python object. |
Nodes
Special type indicating an unconstrained type. |
|
Special type indicating an unconstrained type. |
|
Special type indicating an unconstrained type. |
|
Node correponding to unpackable container. |
|
Leaf node, typically corresponding to a tensor. |
|
Special type indicating an unconstrained type. |