技术栈

pytorch源码

哦豁灬
1 年前
pytorch·深度学习·学习·pytorch源码
pytorch 源码阅读(3)——torch.fxFX 是一个供开发者用来转换 nn.Module 实例的工具包。FX 包含三个主要组件:符号跟踪器(symbolic_traced)、中间表示(intermediate representation,IR)和Python 代码生成(Code generation)。这些组件的应用实例: