参考博客：https://blog.csdn.net/Apikaqiu/article/details/104379960

mean()函数的参数：dim=0,按行求平均值，返回的形状是（1，列数）；dim=1,按列求平均值，返回的形状是（行数，1）,默认不设置dim的时候，返回的是所有元素的平均值。

x=torch.arange(12).view(4,3)

'''

注意：在这里使用的时候转一下类型，否则会报RuntimeError: Can only calculate the mean of floating types. Got Long instead.的错误。

查看了一下x元素类型是torch.int64,根据提示添加一句x=x.float()转为tensor.float32就行

'''

x=x.float()

x_mean=torch.mean(x)

x_mean0=torch.mean(x,dim=0,keepdim=True)

x_mean1=torch.mean(x,dim=1,keepdim=True)

print('x:')

print(x)

print('x_mean0:')

print(x_mean0)

print('x_mean1:')

print(x_mean1)

print('x_mean:')

print(x_mean)

点滴分享，福泽你我！Add oil！