OpTree
latest
Tree Operations
PyTree Node Registry
Integration with
functools
Typing Support
API References
Integration with Third-Party Libraries
OpTree
Index
Edit on GitHub
Index
_
|
A
|
B
|
C
|
D
|
E
|
F
|
I
|
K
|
L
|
M
|
N
|
O
|
P
|
R
|
S
|
T
|
U
|
V
|
W
_
__abstractmethods__ (optree.CustomTreeNode attribute)
__annotations__ (optree.CustomTreeNode attribute)
(optree.PyTree attribute)
(optree.PyTreeKind attribute)
(optree.PyTreeSpec attribute)
__class_getitem__() (optree.PyTree class method)
__copy__() (optree.PyTree method)
__deepcopy__() (optree.PyTree method)
__delattr__() (optree.PyTreeKind method)
(optree.PyTreeSpec method)
__eq__() (optree.PyTreeKind method)
(optree.PyTreeSpec method)
__ge__() (optree.PyTreeKind method)
(optree.PyTreeSpec method)
__getattribute__() (optree.PyTreeKind method)
(optree.PyTreeSpec method)
__getstate__() (optree.PyTreeKind method)
(optree.PyTreeSpec method)
__gt__() (optree.PyTreeKind method)
(optree.PyTreeSpec method)
__hash__() (optree.PyTreeKind method)
(optree.PyTreeSpec method)
__index__() (optree.PyTreeKind method)
__init__() (optree.CustomTreeNode method)
(optree.PyTreeKind method)
(optree.PyTreeSpec method)
__init_subclass__() (optree.PyTree class method)
__int__() (optree.PyTreeKind method)
__le__() (optree.PyTreeKind method)
(optree.PyTreeSpec method)
__len__() (optree.PyTreeSpec method)
__lt__() (optree.PyTreeKind method)
(optree.PyTreeSpec method)
__members__ (optree.PyTreeKind attribute)
__ne__() (optree.PyTreeKind method)
(optree.PyTreeSpec method)
__new__() (optree.PyTree static method)
(optree.PyTreeKind method)
(optree.PyTreeSpec method)
__non_callable_proto_members__ (optree.CustomTreeNode attribute)
__orig_bases__ (optree.CustomTreeNode attribute)
(optree.PyTree attribute)
__parameters__ (optree.CustomTreeNode attribute)
(optree.PyTree attribute)
__protocol_attrs__ (optree.CustomTreeNode attribute)
__setattr__() (optree.PyTreeKind method)
(optree.PyTreeSpec method)
__setstate__() (optree.PyTreeKind method)
(optree.PyTreeSpec method)
__subclasshook__() (optree.CustomTreeNode class method)
A
accessors() (optree.PyTreeSpec method)
all_leaves() (in module optree)
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)
CUSTOM (optree.PyTreeKind attribute)
CustomTreeNode (class in optree)
D
DEFAULTDICT (optree.PyTreeKind attribute)
DEQUE (optree.PyTreeKind attribute)
DICT (optree.PyTreeKind attribute)
E
entries() (optree.PyTreeSpec method)
entry() (optree.PyTreeSpec method)
F
flatten_up_to() (optree.PyTreeSpec method)
I
is_leaf() (optree.PyTreeSpec method)
is_namedtuple() (in module optree)
is_namedtuple_class() (in module optree)
is_namedtuple_instance() (in module optree)
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)
K
kind (optree.PyTreeSpec property)
L
LEAF (optree.PyTreeKind attribute)
LIST (optree.PyTreeKind attribute)
M
MAX_RECURSION_DEPTH (in module optree)
N
name (optree.PyTreeKind property)
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_leaves (optree.PyTreeSpec property)
num_nodes (optree.PyTreeSpec property)
O
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)
register_pytree_node() (in module optree)
register_pytree_node_class() (in module optree)
S
structseq_fields() (in module optree)
STRUCTSEQUENCE (optree.PyTreeKind attribute)
T
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)
(optree.CustomTreeNode method)
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_paths() (in module optree)
tree_ravel() (in module optree.integration.jax)
(in module optree.integration.numpy)
(in module optree.integration.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)
(optree.CustomTreeNode class method)
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_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_ordereddict() (in module optree)
treespec_paths() (in module optree)
treespec_structseq() (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
value (optree.PyTreeKind property)
W
walk() (optree.PyTreeSpec method)
Read the Docs
v: latest
Versions
latest
stable
Downloads
pdf
On Read the Docs
Project Home
Builds