np.argmax函数

当参数是2维时,

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

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

np.argmax的注意点

 

多维时,以此类推。

 

-- over --

相关文章: