PyTorch中形状为[3,224,225]的tensor.mean([1,2])参数[1,2]含义咨询

你的张量形状是torch.Size([3, 224, 225])，PyTorch中张量的维度索引从0开始计数：

• dim=0：维度大小为3，通常可理解为通道数
• dim=1：维度大小为224，比如图像的高度
• dim=2：维度大小为225，比如图像的宽度

调用tensor.mean([1,2])时，[1,2]表示要对第1和第2这两个维度计算均值：

• 针对dim=0上的每一个元素（即每个通道），会把该通道对应的224×225的所有元素求平均值
• 最终得到形状为torch.Size([3])的张量，每个元素对应原张量中一个通道的全局均值，这和你得到的tensor([0.6893, 0.5840, 0.4741])结果完全匹配

额外说明：指定多个维度求均值时，这些维度会被“压缩”，原张量仅保留未指定的维度（这里是dim=0），所以结果的维度数等于原维度数减去指定的维度数量（3-2=1）。

内容的提问来源于stack exchange，提问作者french_fries