tensorflow - 如何使用 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))