PyTorch中RGB图片(x,x,3)变为(3,x,x)

RGB格式的图片，在转为数据的时候，维度是(x,x,3)，但是使用了torchvision.transforms.transforms中的ToTensor方法（将图片转为张量Tensor）之后，维度会自动转为(3,x,x)。

以下是一个例子：

from torchvision.transforms import transforms

x = np.zeros((512, 512, 3))
y = transforms.ToTensor()(x)
print(x.shape)
print(y.shape)

运行结果如下图：