前 5 名的 argmax 的 Tensorflow?
Tensorflow for argmax of top 5?
tf.argmax return 张量的前 1 个。我做了一些研究,但没有找到获得张量前 5 名的好方法(扫描除外)。如果您有更好的方法,请告诉我。谢谢!
您可以使用 tf.nn.top_k
文档 here。
tf.nn.top_k(input, k=5, sorted=True, name=None)
对于您的情况,k=5
,如上所示。
tf.argmax return 张量的前 1 个。我做了一些研究,但没有找到获得张量前 5 名的好方法(扫描除外)。如果您有更好的方法,请告诉我。谢谢!
您可以使用 tf.nn.top_k
文档 here。
tf.nn.top_k(input, k=5, sorted=True, name=None)
对于您的情况,k=5
,如上所示。