在PyTorch中,如果一个张量包括通道数(C)、宽度(W)、高度(H)和批量大小(N),那么它的标准维度顺序是 [N, C, H, W],即:
[N, C, H, W]
N
C
H
W
这种维度顺序在处理图像数据时特别常见,尤其是在使用卷积神经网络(CNN)时。例如,一个包含64张RGB图像(每张图像分辨率为28x28像素)的批次将会有一个形状为 [64, 3, 28, 28] 的张量。
[64, 3, 28, 28]