flex_model.traverse.LeafNode

class flex_model.traverse.LeafNode(val: Any | None = None)

Leaf node, typically corresponding to a tensor.

Note:

Leaf nodes should not hold a ref to the underlying data, only some metadata.

__init__(val: Any | None = None) → None

Methods

__init__([val])