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)

回傳類型: