对 np.transpose（） 的理解

np.transpose(x, (1, 2, 0))

按照指定的axis顺序对array x 进行索引，若x中某个元素的index是（x1, x2, x3），则经上面变换以后其index变为(x2, x3, x0)。易知，np.transpose(x, (0, 1, 2))和 x 相同。 在pytorch中图片的像素点是按照（c, H, W）排列的，我们可以利用上述变换将像素点按照（H, W, c）排布，从而使plt.imshow()可以读取。