前5个argmax的张量流量？

Question

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

Answer 1

您可以使用文档here中的tf.nn.top_k。

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

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