This site does not allow itself to be embedded. You can still read it on the original site — the toolbar below keeps your place in the directory.
简介 Link to heading torch.fx 用于对 torch.nn.Module 做图变换。它包含三部分： 符号跟踪器（symbolic tracer）：用于捕获 module 的语义，它以符号的方式执行 Python 代码（symbolic execution），通过给 module 提供虚假值（Proxies）并记录涉及到的运算； 中间表示（intermediate representation）：IR 是在 tracing 期间所记录算子的图，包含一系列节点（Node），节点代表输入（placeholder）、函数（get_attr, call_function, call_module, call_method）、输出（output），IR 是用 torch.fx 进行图变换（transformation）的基石； Python 代码生成（code…
Comments
Nothing yet. Say the first thing.
Sign in to join the conversation.