frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

fixes import error 'functionalize' from functorch (#163746)

Browse Source 
Fixes #163637

Pull Request resolved: https://github.com/pytorch/pytorch/pull/163746
Approved by: https://github.com/malfet
This commit is contained in:
RiyaP-QA
2025-09-28 23:16:42 +00:00
committed by PyTorch MergeBot
parent d1b3481131
commit bc5a072ebf
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
functorch/experimental/__init__.py
View File

@ -1,5 +1,5 @@
# PyTorch forward-mode is not mature yet
from functorch import functionalize
from torch._functorch.apis import chunk_vmap
from torch._functorch.batch_norm_replacement import replace_all_batch_norm_modules_
from torch._functorch.eager_transforms import hessian, jacfwd, jvp
from torch.func import functionalize