TensorflowでArgsortを実行するには？

Question

第2軸に沿って25×5×5マトリックス（テンソル）をどのように並べ替えることができますか？基本的には、numpyのargsortと同等のテンソルフロー（関数または方法論）を探しています。 np.argsort(matrix, 2)。

Answer 1

おそらくkの値を返すtop_kを使用できます。 kは、ディメンションごとにいくつの値が「トップ」になるかを定義する1Dベクトルになります。あなたの場合、2番目の軸を設定したい場合は、k=[0, 5, 0]を設定してください。

tf.nn.top_k(matrix, k=[0,5,0], sorted=True) 

私はそれを実行しませんでした。 ほしいと思っています。