jax.tree_util.tree_structure#

jax.tree_util.tree_structure(tree, is_leaf=None)[原始碼]#

jax.tree.structure() 的別名。

參數:

  • tree (Any)

  • is_leaf (None | Callable[[Any], bool] | None)

回傳型別:

PyTreeDef