05动手学深度学习（下）

月吟荧静2025-07-30 16:29

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

按特定轴求和

python 复制代码

import torch

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

复制代码

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

python 复制代码

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

复制代码

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

python 复制代码

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

复制代码

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

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