jax.tree_util.tree_all

Contents

jax.tree_util.tree_all#

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

Alias of jax.tree.all().

Parameters:

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

Return type: