jax.tree_util.tree_all

内容

jax.tree_util.tree_all#

jax.tree_util.tree_all(tree, *, is_leaf=None)[source]#

jax.tree.all() 的别名。

参数：

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

返回类型：