jax.tree_util.tree_leaves_with_path# jax.tree_util.tree_leaves_with_path(tree, is_leaf=None)[源代码]# jax.tree.leaves_with_path() 的别名。 参数: tree (Any) is_leaf (Callable[[Any], bool] | None | None) 返回类型: list[tuple[KeyPath, Any]]
