Pytorch中的标准维度顺序

在PyTorch中,如果一个张量包括通道数(C)、宽度(W)、高度(H)和批量大小(N),那么它的标准维度顺序是 [N, C, H, W],即:

  • 第一个维度 N 是批量大小,表示一次处理的样本数。
  • 第二个维度 C 是通道数,如RGB图像中的3个颜色通道。
  • 第三个维度 H 是图像的高度。
  • 第四个维度 W 是图像的宽度。

这种维度顺序在处理图像数据时特别常见,尤其是在使用卷积神经网络(CNN)时。例如,一个包含64张RGB图像(每张图像分辨率为28x28像素)的批次将会有一个形状为 [64, 3, 28, 28] 的张量。

你可能感兴趣的:(PyTorch,pytorch,人工智能,python)