Python numpy 中 keepdims 的含义
keepdims主要用于保持矩阵的二维特性import numpy as npa = np.array([[1,2],[3,4]])# 按行相加，并且保持其二维特性print(np.sum(a, axis=1, keepdims=True))# 按行相加，不保持其二维特性print(np.sum(a, axis=1))输出array([[3], [7]])array([3, 7])
·
keepdims主要用于保持矩阵的二维特性
import numpy as np
a = np.array([[1,2],[3,4]])
# 按行相加，并且保持其二维特性
print(np.sum(a, axis=1, keepdims=True))
# 按行相加，不保持其二维特性
print(np.sum(a, axis=1))输出
array([[3], [7]])
array([3, 7])
有“AI”的1024 = 2048，欢迎大家加入2048 AI社区
更多推荐
48
0- 0
已为社区贡献1条内容
所有评论(0)