rzou 15b1ac3e86 Add torch.func.debug_unwrap (#146528) ...

Use it to unwrap any functorch-wrapped tensor. I don't recommend using
the output in a program since it breaks the semantics of the transforms,
but it seems useful for debugging.

I will note that some people have wanted to get intermediate values out
of an e.g. grad transform, so this might be a way to do that...

Test Plan:
- tests
Pull Request resolved: https://github.com/pytorch/pytorch/pull/146528
Approved by: https://github.com/Chillee

2025-02-06 18:48:09 +00:00

..

__init__.py

Add torch.func.debug_unwrap (#146528)

2025-02-06 18:48:09 +00:00