【numpy】改变矩阵维度，numpy.transpose

在用pytorch处理图像矩阵的时候，我们通常会将三通道RGB图像[256,256,3] 切割成bachsize为128，大小为n*n的小块，得到[128,3,n,n]

测试时，输入图像是大小为[256,256,3]的三通道RGB图像，此时要将图像送入网络，需要经过维度变换处理，将(0,1,2)，变到(2,0,1)。

经过model之后，得到输出图像的维度是[3,256,256]。要想正确显示图像，又需要图像变换处理。将(0,1,2)换成(1,2,0).

此处需要用到numpy的transpose操作.

具体我就不写了，参考这篇博客吧

numpy之转置（transpose）和轴对换https://www.jianshu.com/p/90d4cdfa4000