PyTorch Forums
Script function as target
FX (Functional Transformations)
Yuchen_Wang
(Yuchen Wang)
January 17, 2023, 2:49am
#1
Is it possible to have a torch.jit script function as node target in torch.fx?