我是深度学习和pytorch的新手,我想将模型导出到onnx时发生错误:
RuntimeError: ONNX export failed: Couldn't export operator aten::var
我按照本教程在下面的存储库中添加了对pytorch运算符的支持,但是我做对了。
https://github.com/onnx/tutorials/blob/master/tutorials/PytorchAddExportSupport.md
VariableType.h文件中的运算符行是:
Tensor var(const Tensor & self, IntArrayRef dim, bool unbiased, bool keepdim) const override;