Pytorch常用tensor运算之mean()求平均

dim=0,按行求平均值，返回的形状是（1，列数）

dim=1,按列求平均值，返回的形状是（行数，1）

1 x = torch.randn(2, 2, 2)
2 x

1 tensor([[[-0.7596, -0.4972],
2          [ 0.3271, -0.0415]],
3 
4         [[ 1.0684, -1.1522],
5          [ 0.5555,  0.6117]]])

1 x.mean(-3)

1 tensor([[ 0.1544, -0.8247],
2         [ 0.4413,  0.2851]])

1 x.mean(-3).shape

1 torch.Size([2, 2])