ONNX添加了对torch.var()的导出支持

时间:2019-06-01 10:47:11

标签: pytorch onnx

我是深度学习和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;

0 个答案:

没有答案