PyTree¶
thunder.core.pytree |
pytorch pytree |
---|---|
|
See https://optree.readthedocs.io/en/latest/ops.html#optree.tree_map |
|
See https://optree.readthedocs.io/en/latest/ops.html#optree.tree_flatten |
|
See https://optree.readthedocs.io/en/latest/ops.html#optree.tree_unflatten |