如何在张量流中获得最高n arg max / min？

Question

是否有可能在张量流中获得最高N arg max / min？ 对于最重要的一个，我们有：

tf.argmax(input)

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

获取

Answer 1

这个功能怎么样：tf.nn.top_k？