树工具

在 MLX 中，我们将 Python 树视为任意嵌套的、不含循环的字典、列表和元组的集合。此模块中返回 Python 树的函数将使用默认的 Python dict、list 和 tuple 类型，但它们通常也能处理继承自这些类型的对象。

注意

字典的键应为有效的 Python 标识符。

tree_flatten(tree[, prefix, is_leaf])	将 Python 树展平为键值对列表。
tree_unflatten(tree)	从展平表示重建 Python 树。
tree_map(fn, tree, *rest[, is_leaf])	将函数 fn 应用于 Python 树 tree 的叶子节点，并返回包含结果的新集合。
tree_map_with_path(fn, tree, *rest[, ...])	将函数 fn 应用于 Python 树 tree 的路径和叶子节点，并返回包含结果的新集合。
tree_reduce(fn, tree[, initializer, is_leaf])	对 Python 树的叶子节点应用归约操作。