PyTorch基础（九）----- torch.mean()方法

一、方法详解

含义：顾名思义，返回一个Tensor的均值

torch.mean(input, dim, keepdim=False)

• input：输入的张量

• dim：求均值的维度，若dim = 0，则沿行求均值；若dim=1，则沿列求均值

• keepdim：输出张量是否跟输入张量的另一个维度相同

这个用言语很难说明白，我们直接通过案例来理解掌握！

二、案例

• 案例1

import torch
x = torch.randn(4, 4)
print(x.size())
y = torch.mean(x, dim=0, keepdim=True)
print(y.size())

• 分析

如果我们要沿dim=0这个维度相同，而且keepdim=True，那么输出张量的另一个维度dim=1就跟输入张量相同。

• 运行结果

• 案例2

import torch
x = torch.randn(4, 4)
print(x.size())
y = torch.mean(x, dim=1, keepdim=True)
print(y.size())

• 运行结果