import numpy as np
a = np.array([[1, 2, 3], [4, 5, 6]])
b = np.sum(a, axis=1, keepdims=True)
得到b的结果为:
b = np.sum(a, axis=1)
则b的结果为:
import numpy as np
a = np.array([[1, 2, 3], [4, 5, 6]])
b = np.sum(a, axis=1, keepdims=True)
得到b的结果为:
b = np.sum(a, axis=1)
则b的结果为: