- 基本用法
- 具体使用
mean()是对张量进行求均值的方法，dim用来指定在某个维度上进行求平均。
具体使用- 生成（3，2，1）维的张量
import torch test = torch.randint(1, 4, (3, 2, 1)).float()
- 不指定dim
test1 = test.mean()
即对所有的张量进行求平均，结果为一个数值：
- 指定dim=1
test2 = test.mean(dim=1)
在0,1，2三个维度中，dim=1的维度上求平均，结果中的dim=1的维度会因为已经求了平均而消失，因此结果的shape为（3,1）。
