在操作员级别从深度学习框架导出 ONNX 模型

【问题标题】：Exporting ONNX Model from Deep Learning Frameworks at Operator-level在操作员级别从深度学习框架导出 ONNX 模型
【发布时间】：2021-05-13 17:36:53
【问题描述】：

您好，我有一些关于导出 ONNX 模型的问题。

假设我们有一个来自 PyTorch 的 LSTM 单元。

使用 torch.onnx.export 生成带有 LSTM layer 的 ONNX 模型。

但是，我感兴趣的是它是否可以在 operator 级别生成 ONNX 模型，即 matmul，add。

有办法吗？

如果没有，还有其他方法可以制作运营商级别的ONNX模型吗？

谢谢，杰克

【问题讨论】：

【解决方案1】：

当您使用 torch.onnx.export() 函数将模型从 PyTorch 导出到 onnx 时，它会记录初始模型已使用的所有操作。如here所述。

我们调用torch.onnx.export() 函数。这将执行模型，记录用于计算输出的运算符的踪迹。

所以，是的，它确实在操作员级别生成了 onnx 模型，您甚至可以使用 netron 可视化导出的 .onnx 模型图

如果您仍想使用 onnx 运算符，here 是 ONNX 运算符模式。

【讨论】：