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