tf.argmax()函数其实就是求最大数值所在下标。只是参数不同的时候,表达的列还是行的最大数值。
如tf.argmax(array,0)表示的是求列最大值的索引下标。
tf.argmax(array,1)则表示求行最大值的索引下标。
例子:
array([[[ 1, 2, 3],
[ 4, 3, 2],
[ 2, 2, 5]]
tf.argmax(array,0) #输出的是列最大值的下标[1,1,2]
tf.argmax(array,1) #输出的是行最大值的下标[2,0,2]