tensorflow - 如何在 tensorflow 中获得 top n arg max/min？

Question

是否有可能在 tensorflow 中获得 top N arg max/min？对于第一名，我们有:

tf.argmax(input)

可在 https://www.tensorflow.org/api_docs/python/tf/argmax 获得

Answer 1

这个函数怎么样:tf.nn.top_k ?