這個功能，我在torch中沒有找到解決辦法，但是我們可以將tensor轉換成np.adarray來做

out = np.amax(data,axis = a) #挑選數據data中第a維最大值

返回索引?

idx = np.argmax(data, axis=b)

?

總結

以上是生活随笔為你收集整理的返回多维数组中某一维的最大值的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。