jax.tree_util.treedef_children#

jax.tree_util.treedef_children(treedef)[source]#
Parameters

treedef (PyTreeDef) –

Return type

List[PyTreeDef]