Pytorch使用手册--将 PyTorch 模型导出为 ONNX（专题二十六）

注意
截至 PyTorch 2.1，ONNX 导出器有两个版本。
torch.onnx.dynamo_export 是最新的（仍处于测试阶段）导出器，基于 PyTorch 2.0 发布的 TorchDynamo 技术。
torch.onnx.export 基于 TorchScript 后端，自 PyTorch 1.2.0 起可用。

一、torch.onnx.dynamo_export使用

在 60 分钟入门 中，我们有机会从高层次上了解 PyTorch，并训练了一个小型神经网络来对图像进行分类。在本教程中，我们将在此基础上扩展内容，介绍如何使用 TorchDynamo 和 torch.onnx.dynamo_export ONNX 导出器将 PyTorch 中定义的模型转换为 ONNX 格式。

虽然 PyTorch 非常适合模型的开发迭代，但模型可以通过不同的格式部署到生产环境中，包括 ONNX（开放神经网络交换）！

ONNX 是一种灵活