Jak do Argsort w Tensorflow?

Question

Jak mogę sparować matrycę 25 x 5 x 5 (tensor) wzdłuż 2. osi? Zasadniczo szukam równoważnika tensorflow (funkcji lub metodologii) do argsort numpy, np. np.argsort(matrix, 2).

Answer 1

W twoim przypadku prawdopodobnie możesz użyć wartości top_k, która zwraca najwyższe wartości k. k może być wektorem 1D określającym liczbę wartości "do góry" dla poszczególnych wymiarów. W twoim przypadku, jeśli chcesz, może to zrobić druga oś k=[0, 5, 0].

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

Ja tego nie uruchomiłem. Mam nadzieję, że to pomoże