jax.tree_util.build_tree# jax.tree_util.build_tree(treedef, xs)[source]# Parameters: treedef (PyTreeDef) xs (Any) Return type: Any
