tf.argmax()函數中axis參數理解

tf.argmax是tensorflow用numpy的np.argmax實現的，它能給出某個tensor對象在某一維上的其數據最大值所在的索引值，常用于metric（如acc）的計算

tf.argmax()函數中有個axis參數（軸），該參數能指定按照哪個維度計算。
如 在矩陣的結構中，axis可被設置為0或1，分別表示
0：按列計算，1：行計算

下面是兩個具體的示例，以便于直觀理解（因tf.argmax()的axis的用法與numpy中的argmax相同，為方便起見，本文使用了numpy.argmax()用于示例）

axis=0

當axis=0時，以列為單位，第一列的最大值為索引為3的8，所以第一個值為3，以此類推


axis=1

當axis=1時，以行為單位，第一行的最大值為索引為2的3，所以第一個結果為2


最后編輯于 ：
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明：文章內容（如有圖片或視頻亦包括在內）由作者上傳并發布，文章內容僅代表作者本人觀點，簡書系信息發布平臺，僅提供信息存儲服務。

推薦閱讀更多精彩內容