jax.tree_util.tree_leaves_with_path# jax.tree_util.tree_leaves_with_path(tree, is_leaf=None)[source]# jax.tree.leaves_with_path() 的別名。 參數: tree (Any) is_leaf (Callable[[Any], bool] | None | None) 返回類型: list[tuple[KeyPath, Any]]
