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]