Toggle navigation

.rst .pdf

Warning

This page was created from a pull request (#9655).

jax.tree_util.treedef_children

jax.tree_util.treedef_children(treedef)

previous

jax.tree_util.tree_unflatten

next

jax.tree_util.treedef_is_leaf