torch.mean是对torch.Tensor取均值。具体怎么使用，请看下面的例子

不指定维度求均值

如果不指定维度，是对Tensor里所有元素求均值，最后计算出来的结果是一个标量。

a = torch.Tensor([0, 1, 2, 3, 4, 5]).view(2, 3, 1)
print(a)mean = torch.mean(a)
print(mean, mean.shape)

结果如下：

指定其中一维

如果指定了某一维度，那么剩下的所有元素算均值。
下面的例子是对（2，3，1）Tensor的0维做均值，N=2,即所选的dim，输出为剩下的维度（3，1）
第一个：（0+3)/2=1.5
第二个： (1+4)/2=2.5
第三个： (2+5)/2=3.5

a = torch.Tensor([0, 1, 2, 3, 4, 5]).view(2, 3, 1)
print(a)mean = torch.mean(a, 0)
print(mean, mean.shape)

例子2

如下对dim 1做mean

N=3,即所选的dim，输出为剩下的维度（2，1）
第一个：（0+1+2)/3=1
第二个： (3+4+5)/3=4

a = torch.Tensor([0, 1, 2, 3, 4, 5]).view(2, 3, 1)
print(a)mean = torch.mean(a, 1)
print(mean, mean.shape)

例子3

如下对dim 2做mean

N=1,即所选的dim，输出为剩下的维度（3，2）
[0][0]：（0)/1=0
[0][1]：（1)/1=1
[1][0]：（2)/1=2
[1][1]：（3)/1=3
[2][0]：（4)/1=4
[2][1]：（5)/1=5

a = torch.Tensor([0, 1, 2, 3, 4, 5]).view(2, 3, 1)
print(a)mean = torch.mean(a, 2)
print(mean, mean.shape)

指定多维

其实跟指定一个维度一样，下面如例子

dim=（1，2）
那么N = 2*3 = 6, 输出为（1，）

[0]=(0+1+2+3+4+5)/6 = 2.5

a = torch.Tensor([0, 1, 2, 3, 4, 5]).view(2, 3, 1)
print(a)mean = torch.mean(a, (0,1))
print(mean, mean.shape)

如果需要保持维度，这里的输入为三个维度，那么保持维度的意思就是输出也是三个维度。
那么可以加入keepdim=True，
如下：

a = torch.Tensor([0, 1, 2, 3, 4, 5]).view(2, 3, 1)
print(a)mean = torch.mean(a, (0,1), keepdim=True)
print(mean, mean.shape)

【pytorch】torch.mean的使用相关推荐

PyTorch : torch.nn.xxx 和 torch.nn.functional.xxx
PyTorch : torch.nn.xxx 和 torch.nn.functional.xxx 在写 PyTorch 代码时,我们会发现在 torch.nn.xxx 和 torch.nn.funct ...
PyTorch | torch.linspace()创建均分数列张量 | torch.linspace()如何使用？| torch.linspace()使用方法 | torch.linspace例子
公众号[计算机视觉联盟]后台回复[PyTorch]可以获得独家PyTorch学习教程pdf版通过torch.linspace创建均分数列张量步长=(Start - end)/(Steps - 1 ...
PyTorch | torch.full()使用方法 | torch.full()如何使用？ torch.full()例子说明 | 通过torch.full创建全相同的张量
公众号[计算机视觉联盟]后台回复[PyTorch]可以获得独家PyTorch学习教程pdf版举例子说明torch.full()使用方法: t = torch.full((3,3),10)print( ...
PyTorch | torch.zeros()如何使用？torch.zeros使用方法 | torch.zeros()例子
公众号[计算机视觉联盟]后台回复[PyTorch]可以获得独家PyTorch学习教程pdf版还是用实际例子比较明显 out_t = torch.tensor(([1]))t = torch.zero ...
PyTorch | torch.randperm()使用方法
公众号[计算机视觉联盟]后台回复[PyTorch]可以获得PyTorch学习教程pdf版返回一个0~n-1的数组,随机打散的 t = torch.randperm(8) 结果: tensor([5, ...
PytorchRuntimeError: inconsistent tensor sizes at /pytorch/torch/lib/TH/generic/THTensorMath.c:2709
1.pytorch在得到dataset时没有问题,得到DataLoader时也没有问题,以batch size = 512遍历数据时,出现了: RuntimeError: inconsistent t ...
如何使用PyTorch torch.max（）
In this article, we'll take a look at using the PyTorch torch.max() function. 在本文中,我们将介绍如何使用PyTorch ...
pytorch torch.Tensor.numpy()（从张量创建一个numpy数组，数组和张量共享相同内存）
https://pytorch.org/docs/1.1.0/tensors.html?highlight=numpy#torch.Tensor.numpy numpy() → numpy.ndarr ...
pytorch torch.from_numpy()（从numpy数组创建一个张量，数组和张量共享相同内存）
https://pytorch.org/docs/1.1.0/torch.html?highlight=numpy#torch.from_numpy torch.from_numpy(ndarray) ...
pytorch torch.item()（返回此张量的值作为标准Python数字。这仅适用于具有一个元素的张量。）
https://pytorch.org/docs/1.1.0/tensors.html?highlight=item#torch.Tensor.item item() → number Returns ...

【pytorch】torch.mean的使用

不指定维度求均值

指定其中一维

例子2

例子3

指定多维

【pytorch】torch.mean的使用相关推荐

最新文章

热门文章