---
title: PyTree
canonical: 'https://lightning.ai/docs/thunder/stable/api-reference/core/pytree'
---
# PyTree {#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 |

---

**Navigation**

- Parent: [thunder.core](/docs/thunder/0.2.6/api-reference/core.md)