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)

返回类型: