转换成onnx的注意事项：
1）对于任意用到shape、size返回数的时候，避免直接使用tensor.size()的返回值，而是加上int转换，例如tensor.view(-1,int(tensor.size(1)));
2)对于nn.upsample或者nn,functional.interpolate函数，使用scale_factor指定倍数,而不是使用size参数指定大小(对于无法使用倍数的，可以使用size来指定大小);
3）对于reshape、view操作时，batch维度用-1来指定，其他维度可以通过计算得到（方便动态batch）；
4) torch.cat改变维度后，在对其数据进行softmax；不然，会出现下面例子的错误（具体原因希望有大佬能解释一下）。
参考文献：
tensorRT高性能部署