版权声明：本文为博主原创文章，转载请标明出处，蟹蟹 https://blog.csdn.net/fly_wt/article/details/85843787
tf.argmax(vector, 1)：返回的是vector中的最大值的索引号，如果vector是一个向量，那就返回一个值，如果是一个矩阵，那就返回一个向量，这个向量的每一个维度都是相对应矩阵行的最大值元素的索引号。
import tensorflow as tf
A = tf.constant([[1,2,3,4,5]])
B = tf.constant([[1,2,3],[6,5,4]])
with tf.Session() as sess:
print(sess.run(tf.argmax(A,1)))
print(sess.run(tf.argmax(B,1)))
输出
[4]
[2 0]