PyTorch笔记——FX
官方文档链接:https://pytorch.org/docs/master/fx.html#
概述
FX是供开发人员用于转换nn.Module实例的工具包。FX由三个主要组件组成:符号追踪:symbolic tracer, 中间层表示:intermediate representation, Python代码生成:Python code generation。这些组件的运行演示:
import torch
# Simple module for demonstration
class MyModule(t
原创
2021-10-22 19:55:58 ·
1616 阅读 ·
0 评论