tensorflow - 前 5 名的 argmax 的 Tensorflow？

Question

tf.argmax 返回张量的前 1 个。我做了一些研究，但没有找到获得张量前 5 名的好方法（除了扫描）。如果您有更好的方法，请告诉我。谢谢！

Answer 1

您可以使用此处tf.nn.top_k文档中的内容。

tf.nn.top_k(input, k=5, sorted=True, name=None)

对于您的情况，k=5，如上所示。