OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 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
Show Stats 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 # PyTorch forward-mode is not mature yet
from functorch import functionalize
from torch._functorch.apis import chunk_vmap from torch._functorch.apis import chunk_vmap
from torch._functorch.batch_norm_replacement import replace_all_batch_norm_modules_ from torch._functorch.batch_norm_replacement import replace_all_batch_norm_modules_
from torch._functorch.eager_transforms import hessian, jacfwd, jvp from torch._functorch.eager_transforms import hessian, jacfwd, jvp
from torch.func import functionalize