pytorch文档地址与ONNX算子文档

RuntimeError: ONNX export failed: Couldn't export operator aten::upsample_bilinear2d

当pytorch转换onnx时候，出现这一类错误，一般就是，onnx还没有该类算子

    def forward(self, x, out_size):
        x = self.conv1(x)
        x = self.dropout(x)
        x = self.conv2(x)
        #x = F.interpolate(x, size=out_size, mode="bilinear", align_corners=True)
        x = F.upsample_nearest(x, size=out_size)
        return x

 https://pytorch.org/docs/stable/nn.html?highlight=upsample#torch.nn.functional.upsample_nearest

替换一些onnx可能有的算子

---

---

https://pytorch.apachecn.org/

https://github.com/onnx/onnx/blob/master/docs/Operators.md