簡體   English   中英

如何在張量流中獲得前n個arg max / min?

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

是否有可能在張量流中獲得最高的N arg max / min? 對於第一名,我們有:

tf.argmax(input)

可在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