【pytorch】如何理解tensor的sum()函数（axis、keepdims）

函数： sum(axis=0,keepdim=True)
在二维向量中，axis=0表示行，axis=1表示列。
三维向量中（类比书本）axis=0表示页，axis=1表示行，axis=2表示列。
keepdims为True，表示求和前后shape不变，keepdims默认为False。

1.axis参数

axis=0，就去掉axis=0的维度。
例子（二维）：原来shape为[5,4]，–>（求和后）shape为[4]。可以理解为压扁axis=0。

sum(axis=0)

2.keepdims参数

例子：三维，原来shape为[2,5,4]–>（求和后）shape为[2,1,4]

sum(axis=1,keepdim=True)

3.看个代码

4.参考

李沐老师-动手学深度学习