TensorFlow函數（七）tf.argmax()

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]