tf.argmax()

tf.argmax()

函数原型：

tf.argmax(input,axis=None,name=None,dimension=None,  output_type=tf.int64)

作用：

• 返回最大值索引

例子：

import tensorflow as tf
b = tf.constant([[1,2,3],[3,2,1],[4,5,6],[6,5,4]])
c=tf.argmax(b, 0)#按列
c1=tf.argmax(b, 1)#按行
with tf.Session() as sess:
    print(sess.run([c,c1]))
    #array([3, 2, 2], dtype=int64), array([2, 0, 2, 0], dtype=int64)