torch.mean(input) 输出input 各个元素的的均值,不指定任何参数就是所有元素的算术平均值,指定参数可以计算每一行或者 每一列的算术平均数
a = torch.randn(1, 3)
tensor([[ 0.2294, -0.5481, 1.3288]])
torch.mean(a)
tensor(0.3367)
torch.mean(input) 输出input 各个元素的的均值,不指定任何参数就是所有元素的算术平均值,指定参数可以计算每一行或者 每一列的算术平均数
a = torch.randn(1, 3)
tensor([[ 0.2294, -0.5481, 1.3288]])
torch.mean(a)
tensor(0.3367)