1.文字解释
按特定轴求和，即设定axis的取值。
axis=0表示将原矩阵的第0维度丢弃
axis=1表示将原矩阵的第1维度丢弃
axis=i表示将原矩阵的第i维度丢弃。
具体的可参阅下图：
2.python 帮助理解
import torch
a = torch.ones((2,5,4))
a
#运行结果
tensor([[[1., 1., 1., 1.],
[1., 1., 1., 1.],
[1., 1., 1., 1.],
[1., 1., 1., 1.],
[1., 1., 1., 1.]],
[[1., 1., 1., 1.],
[1., 1., 1., 1.],
[1., 1., 1., 1.],
[1., 1., 1., 1.],
[1., 1., 1., 1.]]])axis=0
axis=1
axis=[0,2]
注意！！
如果设定keepdim=True,则相应的axis=i的第2个维度不再是直接丢弃（消失），而是变成了1，具体可参见下图：
python 帮助理解
axis=0,keepdim=True
axis=1,keepdim=True
axis=[0,2],keepdim=True
不再是第0维度，第2维度丢弃，维度变成了1
文章出处登录后可见！
已经登录？立即刷新
