pytorch中的图像是cwh嘛？

在PyTorch中，图像通常表示为一个四维张量，维度顺序为：[batch_size, channel, height, width]，其中：

• batch_size：表示输入的图像数量；
• channel：表示图像的通道数，例如RGB图像有3个通道，灰度图像只有1个通道；
• height：表示图像的高度；
• width：表示图像的宽度。

因此，可以认为PyTorch中的图像表示为channel-first的形式，即通道数在第二个维度上。