TensorFlow函数（七）tf.argmax()

tf.argmax(input, dimension, name=None)

参数：

• input：输入数据
• dimension：按某维度查找。

　　　　dimension=0：按列查找；

　　　　dimension=1：按行查找；

返回：

• 最大值的下标

 a = tf.constant([1.,2.,3.,0.,9.,])
 b = tf.constant([[1,2,3],[3,2,1],[4,5,6],[6,5,4]])
 with tf.Session() as sess:
     sess.run(tf.argmax(a, 0))
 with tf.Session() as sess:
     sess.run(tf.argmax(b, 0))
 with tf.Session() as sess:
     sess.run(tf.argmax(b, 1))

输出：

4

输出：

[3, 2, 2]

输出：

[2, 0 ,2, 0]