Index

_

__class_getitem__() (optree.PyTree class method) __contains__() (optree.PyTree method) __eq__() (optree.PyTreeSpec method) __format__() (optree.PyTreeKind method) __ge__() (optree.PyTreeSpec method) __getattr__() (optree.PyTree method) __getitem__() (optree.PyTree method) __gt__() (optree.PyTreeSpec method) __hash__() (optree.PyTreeSpec method) __init__() (optree.CustomTreeNode method) (optree.PyTreeSpec method)

__init_subclass__() (optree.PyTree class method) __iter__() (optree.PyTree method) __le__() (optree.PyTreeSpec method) __len__() (optree.PyTree method) (optree.PyTreeSpec method) __lt__() (optree.PyTreeSpec method) __ne__() (optree.PyTreeSpec method) __new__() (optree.PyTree static method) (optree.PyTreeKind method) __tree_flatten__() (optree.CustomTreeNode method) __tree_unflatten__() (optree.CustomTreeNode class method)

A

accessors() (optree.PyTreeSpec method)

all_leaves() (in module optree) AttrsEntry (class in optree.integrations.attrs)

B

broadcast_common() (in module optree)

broadcast_prefix() (in module optree) broadcast_to_common_suffix() (optree.PyTreeSpec method)

C

child() (optree.PyTreeSpec method) children() (optree.PyTreeSpec method) compose() (optree.PyTreeSpec method)

count() (optree.PyTree method) CUSTOM (optree.PyTreeKind attribute) CustomTreeNode (class in optree)

D

dataclass() (in module optree.dataclasses) DEFAULTDICT (optree.PyTreeKind attribute) define() (in module optree.integrations.attrs)

DEQUE (optree.PyTreeKind attribute) DICT (optree.PyTreeKind attribute) dict_insertion_ordered() (in module optree)

E

entries() (optree.PyTreeSpec method)

entry() (optree.PyTreeSpec method)

F

field() (in module optree.dataclasses) (in module optree.integrations.attrs)

flatten_up_to() (optree.PyTreeSpec method) frozen() (in module optree.integrations.attrs)

G

get() (optree.PyTree method)

I

index() (optree.PyTree method) is_leaf() (optree.PyTreeSpec method) is_namedtuple() (in module optree) is_namedtuple_class() (in module optree) is_namedtuple_instance() (in module optree) is_one_level() (optree.PyTreeSpec method)

is_prefix() (optree.PyTreeSpec method) is_structseq() (in module optree) is_structseq_class() (in module optree) is_structseq_instance() (in module optree) is_suffix() (optree.PyTreeSpec method) items() (optree.PyTree method)

K

keys() (optree.PyTree method)

kind (optree.PyTreeSpec property)

L

LEAF (optree.PyTreeKind attribute)

LIST (optree.PyTreeKind attribute)

M

make_class() (in module optree.integrations.attrs) make_dataclass() (in module optree.dataclasses) MAX_RECURSION_DEPTH (in module optree) module optree.dataclasses optree.functools optree.pytree optree.treespec

mutable (in module optree.integrations.attrs)

N

NAMEDTUPLE (optree.PyTreeKind attribute) namedtuple_fields() (in module optree) namespace (optree.PyTreeSpec property) NONE (optree.PyTreeKind attribute) NONE_IS_LEAF (in module optree)

none_is_leaf (optree.PyTreeSpec property) NONE_IS_NODE (in module optree) num_children (optree.PyTreeSpec property) NUM_KINDS (optree.PyTreeKind attribute) num_leaves (optree.PyTreeSpec property) num_nodes (optree.PyTreeSpec property)

O

one_level() (optree.PyTreeSpec method) optree.dataclasses module optree.functools module

optree.pytree module optree.treespec module ORDEREDDICT (optree.PyTreeKind attribute)

P

partial (class in optree.functools) paths() (optree.PyTreeSpec method) prefix_errors() (in module optree) PyTree (class in optree)

PyTreeDef (in module optree) PyTreeKind (class in optree) PyTreeSpec (class in optree) PyTreeTypeVar() (in module optree)

R

reduce() (in module optree.functools) reexport() (in module optree.pytree)

register_node() (in module optree.integrations.attrs) register_pytree_node() (in module optree) register_pytree_node_class() (in module optree)

S

structseq_fields() (in module optree)

STRUCTSEQUENCE (optree.PyTreeKind attribute)

T

transform() (optree.PyTreeSpec method) traverse() (optree.PyTreeSpec method) tree_accessors() (in module optree) tree_all() (in module optree) tree_any() (in module optree) tree_broadcast_common() (in module optree) tree_broadcast_map() (in module optree) tree_broadcast_map_with_accessor() (in module optree) tree_broadcast_map_with_path() (in module optree) tree_broadcast_prefix() (in module optree) tree_flatten() (in module optree) tree_flatten_one_level() (in module optree) tree_flatten_with_accessor() (in module optree) tree_flatten_with_path() (in module optree) tree_is_leaf() (in module optree) tree_iter() (in module optree) tree_leaves() (in module optree) tree_map() (in module optree) tree_map_() (in module optree) tree_map_with_accessor() (in module optree) tree_map_with_accessor_() (in module optree) tree_map_with_path() (in module optree) tree_map_with_path_() (in module optree) tree_max() (in module optree) tree_min() (in module optree) tree_partition() (in module optree) tree_paths() (in module optree) tree_ravel() (in module optree.integrations.jax) (in module optree.integrations.numpy) (in module optree.integrations.torch) tree_reduce() (in module optree) tree_replace_nones() (in module optree)

tree_structure() (in module optree) tree_sum() (in module optree) tree_transpose() (in module optree) tree_transpose_map() (in module optree) tree_transpose_map_with_accessor() (in module optree) tree_transpose_map_with_path() (in module optree) tree_unflatten() (in module optree) treespec_accessors() (in module optree) treespec_child() (in module optree) treespec_children() (in module optree) treespec_defaultdict() (in module optree) treespec_deque() (in module optree) treespec_dict() (in module optree) treespec_entries() (in module optree) treespec_entry() (in module optree) treespec_from_collection() (in module optree) treespec_is_leaf() (in module optree) treespec_is_one_level() (in module optree) treespec_is_prefix() (in module optree) treespec_is_strict_leaf() (in module optree) treespec_is_suffix() (in module optree) treespec_leaf() (in module optree) treespec_list() (in module optree) treespec_namedtuple() (in module optree) treespec_none() (in module optree) treespec_one_level() (in module optree) treespec_ordereddict() (in module optree) treespec_paths() (in module optree) treespec_structseq() (in module optree) treespec_transform() (in module optree) treespec_tuple() (in module optree) TUPLE (optree.PyTreeKind attribute) type (optree.PyTreeSpec property)

U

unflatten() (optree.PyTreeSpec method)

unregister_pytree_node() (in module optree)

V

values() (optree.PyTree method)

W

walk() (optree.PyTreeSpec method)