np.argmax的注意点

np.argmax函数

当参数是2维时,

axis=-1,代表纵轴,即选择纵轴方向最大的元素下标(即每一行的最大值

axis=0,代表横轴,即选择横轴方向最大的元素下标(即每一列的最大值

多维时,以此类推。

-- over --

猜你喜欢

转载自blog.csdn.net/qm5132/article/details/83379870
np
今日推荐