tf.argmax()

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/fyq201749/article/details/81986926

tf.argmax() 返回矩阵横列或者纵列的最大值的坐标，取决于第二个参数
参数：

tf.argmax(
    input,
    axis=None,//0是纵列 1是横列
    name=None,
    dimension=None,
    output_type=tf.int64
)

使用案例：

import tensorflow as tf
x = tf.constant([[0, 2, 1], [2, 1, 3]])
sess = tf.Session()
print(sess.run(tf.argmax(x, 0)))//[1 0 1]
print(sess.run(tf.argmax(x, 1)))//[1 2]