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)

運行結果如下圖：