简体   繁体   English

如何在张量流中获得前n个arg max / min?

[英]How to get top n arg max/min in tensorflow?

Is it possible to get top N arg max/min in tensorflow? 是否有可能在张量流中获得最高的N arg max / min? For top one we have: 对于第一名,我们有:

tf.argmax(input)

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

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

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM