Tensorflow爲前5名的argmax？

Question

tf.argmax返回張量的前1位。我做了一些研究，沒有找到一種好的方法（掃描除外）來獲得張量的前5位。如果你有更好的方法，請告訴我。謝謝！

Answer 1

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

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

對於您的情況，k=5，如上所示。