线性代数 动手学深度学习 pytorch

axis = 0按照某个维度求和，就是去掉了某个维度，比如 shape: [5, 4] axis = 0, 相当于去掉维度0，也就是5，剩下维度4 sum: [4].
deepdims=True 表示保持维度，但是去掉的维度axis 都为1. 比如 shape: [5, 4] axis = 0, deepdims=True , 结果就是sum: [1, 4] 例子：
import torch

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

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

a.sum().shape

torch.Size([])

a.sum(axis=1)

tensor([[5., 5., 5., 5.],
[5., 5., 5., 5.]])

a.sum(axis=1).shape

torch.Size

线性代数 动手学深度学习 pytorch最先出现在Python成神之路。