如何使用tf.argmax

Question

我想测试tf.argmax（）的功能，但是当我运行代码时，我遇到了一个错误。这是我的代码

import tensorflow as tf

a=tf.argmax([1,0,0],1)
with tf.Session() as sess:
    print(sess.run(a))

我的环境是：python3 + tf1.3
代码有什么问题？

Answer 1

在tensorflow中，使用argmax（）和argmin（）函数来查找向量中的最大和最小索引值。代码的问题在于，您将轴参数指定为“ 1”，这意味着您要在二维数组中进行搜索。请检查以下链接：https://www.dotnetperls.com/arg-max-tensorflow

import tensorflow as tf

a=tf.argmax([1,0,0],0)
with tf.Session() as sess:
    print(sess.run(a))