應用 keepdims 時使用 np.argmax 時出錯

Question

我正在運行我的 Python 代碼並在 keepdims 上收到此錯誤：

在此處輸入圖像描述

這是代碼：

在此處輸入圖像描述

幾天前在我的計算機上運行此命令效果很好，但我運行了其他代碼等之后可能已經做了一些事情。

它可以在 amax 上編寫 keepdims，而不是在 argmax 上。 我的朋友現在在她的電腦上運行了同樣的程序，即使代碼相同，這個錯誤也沒有出現。 我嘗試卸載並重新安裝 anaconda 但它沒有改變它。 不確定是否還有其他我必須下載的東西或發生了什么。

Answer 1

對於數組x ，復制np.argmax(x, axis=0, keepdims=True)行為的一種簡單方法是np.argmax(x, axis=0)[np.newaxis, ...] 。 請注意，這是專門針對axis=0的情況。

其他替代方案包括np.expand_dims(np.argmax(x, axis=0), 0)和np.argmax(x, axis=0).reshape((1,) + x.shape[1:]) 。

對於任意軸k ， np.expand_dims(np.argmax(x, axis=k), k)將起作用。