axis其实很简单啦,就是在这个维度内的元素做对应的运算,然后去除掉这个维度:
举例:
axis = 0表示对最外层[]里的最大单位块做块与块之间的运算,同时移除最外层[]: [4 5 6 7]
axis= 1表示对第二外层[]里的最大单位块做块与块之间的运算,同时移除第二外层[]
res = np.array([[[2],[3],[4],[-2]], [[2],[3],[4],[-2]]])
# print(res[range(2), np.array([2, 2])])
res = np.array([[1,2,3,4], [3,3,3,3]])
print(np.sum(res, axis=0)) # axis = 0表示对最外层[]里的最大单位块做块与块之间的运算,同时移除最外层[]: [4 5 6 7]
# axis= 1表示对第二外层[]里的最大单位块做块与块之间的运算,同时移除第二外层[]:
print(np.sum(res, axis=1)) # 1 2 3 4 就是不同的块 [10 12]
np.array([[[1,2],[3,4]],[[11,12],[13,14]]])
a.sum(axis = 1)
array([[ 4, 6], [24, 26]])
a.sum(axis=2)array([[3,7], [23,27]])