トップ5のargmaxのためのテンソルフロー？

Question

tf.argmaxはテンソルのトップ1を返します。私はいくつかの研究を行い、テンソルのトップ5を得るための良い方法（スキャン以外）を見つけられませんでした。あなたがより良いアプローチをしているかどうか教えてください。ありがとう！

Answer 1

tf.nn.top_kは、hereのマニュアルに記載されています。

tf.nn.top_k(input, k=5, sorted=True, name=None) 

あなたのケースでは、上記のようにk=5です。