树工具

在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树的叶子节点应用归约操作。