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

sal*_*jad 2 tensorflow

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

tf.argmax(input)
Run Code Online (Sandbox Code Playgroud)

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

Dmi*_*kiy 6

这个功能怎么样:tf.nn.top_k