np.argmax的注意點

np.argmax函數 當參數是2維時, axis=-1,代表縱軸,即選擇縱軸方向最大的元素下標(即每一行的最大值) axis=0,代表橫軸,即選擇橫軸方向最大的元素下標(即每一列的最大值)   多維時,以此類推。   -- over --
相關文章
相關標籤/搜索