如何獲得張量最大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？