torch.onnx.export( 参数详细解析

`torch.onnx.export`是PyTorch提供的一个函数，用于将PyTorch模型转换为ONNX (Open Neural Network Exchange) 格式。该函数的基本语法如下：

torch.onnx.export(model,
                  inputs,
                  export_path,
                  input_names=['input_0'],
                  output_names=['output_0'],
                  dynamic_axes=None,
                  opset_version=11,
                  do_constant_folding=True,
                  verbose=False,
                  training=torch.onnx.TrainingMode.EVAL,
                  _export_params=True,
                  operator_export_type=torch.onnx.OperatorExportTypes.ONNX)

参数解释：

1. `model`: 要转换的PyTorch模型实例。
2. `inputs`: 模型输入数据的一组张量或一个包含所有输入的列表。这些数据应该与模型训练时使用的输入一致。
3. `export_path`: 输出文件的路径，ONNX模型将会保存到这个路径。
4. `input_names` 和 `output_names`: 分别是输入和输出张量在ONNX图中的名称列表。
5. `dynamic_axes` (可选): 如果模型中的轴大小未知或动态变化，可以提供一个字典映射轴名到一个表示其变动范围的对象。
6. `opset_version`: ONNX版本，指定导出的模型所支持的最低ONNX版本。
7. `do_constant_folding` (可选): 是否进行常数折叠优化。
8. `verbose` (可选): 是否显示转换过程的日志信息。
9. `training`: 模型的训练模式，一般设为`TrainingMode.EVAL`以保证在评估状态下的转换。
10. `_export_params` (可选): 是否导出模型的参数。默认值为True。
11. `operator_export_type` (可选): 控制导出时对某些非标准操作的行为，如上面提到的`ONNX_ATEN_FALLBACK`。

使用这个函数时，你需要确保你的模型在评估模式下，且所有需要的库已经安装了。