【pytorch】——报错 exporting the operator pixel_unshuffle to onnx opset version 11 is not supported

pytorch, onnx

问题：

pytorch的模型转onnx的时候，报错：
exporting the operator pixel_unshuffle to onnx opset version 11 is not supported
但是用pytorch.jit.trace则是可以导出的。

解决方案

• 自己写一个pixel_unshuffle导出到onnx的插件
• 使用pytorch jit.trace追踪出来的模型。