jax.tree_util.tree_leaves# jax.tree_util.tree_leaves(tree, is_leaf=None)[原始碼]# jax.tree.leaves() 的別名。 參數: tree (Any) is_leaf (Callable[[Any], bool] | None | None) 返回類型: list[Leaf]
