pytorch之torch.mean()

基本用法

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）。