tensorflow 如何使用tf.argmax

ckx4rj1h 于 2023-01-13 发布在其他

关注(0)|答案(1)|浏览(160)

我想测试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。
密码有什么问题？

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))