tensorflow argmax

tf.argmax(vector, 1)：返回的是vector中的最大值的索引号，如果vector是一个向量，那就返回一个值，如果是一个矩阵，那就返回一个向量，这个向量的每一个维度都是相对应矩阵行的最大值元素的索引号。

[python]  view plain  copy

1. import tensorflow as tf  
2. import numpy as np  
3.   
4. A = [[1,3,4,5,6]]  
5. B = [[1,3,4], [2,4,1]]  
6.   
7. with tf.Session() as sess:  
8.     print(sess.run(tf.argmax(A, 1)))  
9.     print(sess.run(tf.argmax(B, 1)))  

输出：

[4]
[2 1]