05动手学深度学习（下）

按特定轴求和

看到后面发现沐神单独分开是为了解释其中的线性代数部分中按特定轴求和部分的理解。

按特定轴求和

import torch

a=torch.ones((2,5,4))
a.shape

torch.Size([2, 5, 4])

a.sum(axis=[0,2])

tensor([8., 8., 8., 8., 8.])

a.sum(axis=[0,2],keepdims=True).shape

torch.Size([1, 5, 1])

上面的都可以换，比如axis可以换成0、1什么的，大家都可以尝试一下，就比较好理解具体是怎样求和的。