Pytorch中的标准维度顺序

发布时间:2024年01月14日

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

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

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

文章来源:https://blog.csdn.net/qq_61980594/article/details/135587323
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。