jax.tree_util.tree_all

目录

jax.tree_util.tree_all#

jax.tree_util.tree_all(tree, *, is_leaf=None)[源代码]#

是 jax.tree.all() 的别名。

参数:

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

返回类型：