PyTree

pytree API map

thunder.core.pytree	pytorch pytree
tree_map()	See https://optree.readthedocs.io/en/latest/ops.html#optree.tree_map
tree_flatten()	See https://optree.readthedocs.io/en/latest/ops.html#optree.tree_flatten
tree_unflatten()	See https://optree.readthedocs.io/en/latest/ops.html#optree.tree_unflatten