6
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).Jak do Argsort w Tensorflow?
A
Odpowiedz
7
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