Tensorflow에서 Argsort하는 방법?

Question

두 번째 축을 따라 25 x 5 x 5 매트릭스 (텐서)를 어떻게 배열 할 수 있습니까? 본질적으로, 나는 numpy의 argsort에 대한 tensorflow의 동등한 (함수 또는 방법론)을 찾고있다. np.argsort(matrix, 2).

Answer 1

아마도 k 값을 반환하는 top_k을 사용할 수 있습니다. k은 치수 당 '최고'까지 몇 개의 값을 정의하는 1D 벡터가 될 수 있습니다. 귀하의 경우, 두 번째 축을 원하면 k=[0, 5, 0]으로 설정하십시오.

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

나는 그것을 실행하지 않았다. 희망이 있습니다.