torch.onnx.errors.UnsupportedOperatorError: Exporting the operator ‘aten::unflatten‘ to ONNX opset

项目场景：

转onnx时出现报错提示：torch.onnx.errors.UnsupportedOperatorError: Exporting the operator ‘aten::unflatten’ to ONNX opset version 11 is not supported. Please feel free to request support or submit a pull request on PyTorch GitHub: https://github.com/pytorch/pytorch/issues.

原因分析：

这个错误表明在尝试将 PyTorch 模型导出为 ONNX 格式时，遇到了不受支持的操作符 “aten::unflatten”。这意味着在当前的 ONNX opset 版本（版本 11）中，不支持将 “aten::unflatten” 运算导出为 ONNX 格式。

---

解决方案：

要解决这个问题。

1. 降低ONNX opset版本: 你可以尝试使用较低版本的 ONNX opset（如版本 10），因为较低版本的 opset 可能支持更少的运算，但也可能会限制模型的某些功能。你可以在导出时指定 opset 的版本，例如：

   torch.onnx.export(model, input, "model.onnx", opset_version=10)
   

2. 检查模型结构：如果 “aten::unflatten” 运算在你的模型中被广泛使用，并且目前无法导出为 ONNX 格式，那么你可能需要考虑修改模型结构，以使用 ONNX 支持的运算来替代 “aten::unflatten”。

3. 使用低版本的pytorch，之前使用的2.0.0+cu117，改为1.9.0+cu117后转onnx正常。