TorchDynamo初探Torch

作者｜strint

1

概要

torch.fx是PyTorch官方发布的Python到Python的代码变换工具。如果你想做Torch代码变换，torch.fx是首选工具。

torch.fx会将Torch代码trace成6种基础的node组成的graph，基于这个graph可以方便的做各种变换，变换后的graph可以再生成torch代码（一个nn.Module），然后像普通的nn.Module一样去执行。

torch2.0新发布的torch.

转载请注明：http://www.aierlanlan.com/grrz/7277.html