tf.argmax(input, dimension, name=None)
参数：
- input：输入数据
- dimension：按某维度查找。
dimension=0：按列查找；
dimension=1：按行查找；
返回：
- 最大值的下标
1 a = tf.constant([1.,2.,3.,0.,9.,]) 2 b = tf.constant([[1,2,3],[3,2,1],[4,5,6],[6,5,4]]) 3 with tf.Session() as sess: 4 sess.run(tf.argmax(a, 0)) 5 with tf.Session() as sess: 6 sess.run(tf.argmax(b, 0)) 7 with tf.Session() as sess: 8 sess.run(tf.argmax(b, 1))
输出：
4
输出：
[3, 2, 2]
输出：
[2, 0 ,2, 0]