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


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

推薦閱讀更多精彩內容