【Pytorch基础】二维/三维情况下 torch.mean()函数使用

1 前言

  在这篇介绍中【Pytorch】张量的维度/轴/dim的理解已经解释了dim的用法，下面具体看torch.mean()的用例。

2 一维二维

# 二维 
a = torch.arange(6) * 1.
b = a.reshape(2, 3)
print(b.shape)
print(b)
print(b.mean(dim=0).shape)
print(b.mean(dim=0))
print(b.mean(dim=1).shape)
print(b.mean(dim=1))
1
2
3
4
5
6
7
8
9

3 参考文献

[1]torch.mean()
[2]PyTorch基础（九）----- torch.mean()方法
[3]orch.mean()、tensor.mean() 用法及维度的理解(二维、三维)
[4]torch.mean()