前 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,如上所示。