快捷方式

FXE0008:fx-node-to-onnx

此诊断跟踪从 FX 节点到 ONNX 算子 的转换过程。

将 FX 节点转换为 ONNX 节点的过程涉及处理六种不同的节点类型

  1. placeholder:表示模块输入,映射到 ONNX 图输入。

  2. call_module:象征对子模块的调用,映射到 ONNX

  3. call_method:象征方法调用。尚未实现。

  4. call_function:象征函数调用。核心 ATen 预计为函数调用目标。从 ATen 到 ONNX 的映射由 ONNXScript torchlib 实现。此 指南 展示了如何为 call_function FX 节点编写和注册自定义符号函数。

  5. get_attr:表示当前模块中的属性访问。映射到 ONNX 图初始化程序。

  6. output:表示模块的输出。映射到 ONNX 图输出。

要深入了解每种节点类型是如何转换的,请参阅 FxOnnxInterpreter 中的实现细节。

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取初学者和高级开发人员的深度教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源