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