526互联

tensor.mean()

发布时间 2023-11-20 15:53:14作者: MrNobody

转载Pytorch：tensor.mean()和tensor.sum()_三世的博客-CSDN博客

import torch
x = torch.arange(24).view(2, 3, 4).float()
y = x.mean(0)
print("x.shape:", x.shape)
print("x:")
print(x)
print("y.shape:", y.shape)
print("y:")
print(y)

img

如果对i维度求均值，则就是拿出不同i维度的值求均值

如对2求均值，就是对行求均值，即对不同行求和，然后除行数

如果是多个 维度一起求均值

则先求低维度，再求搞维度

import torch
x = torch.arange(24).view(2, 3, 4).float()
y = x.mean([0,1])
print("x.shape:", x.shape)
print("x:")
print(x)
print("y.shape:", y.shape)
print("y:")
print(y)

img

先求dim=1，再求dim=0