PyTree

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

Was this page helpful?