Есть ли функция TF, аналогичная данным ранга SciPy

Question

Вопрос:

SciPy имеет функцию , rankdata, которая возвращает ранг каждого индекса матрицы.

Пример

    Values          Ranked
-------------   ------------
[0, 4, 2, 6]     [0, 2, 1, 3]
[7, 5, 1, 0] --> [3, 2, 1, 0]
[1, 8, 1, 0]     [1, 3, 1, 0]

Есть ли способ сделать это с Tensorflow? Мне нужно, чтобы графический процессор ускорил функцию данных ранга на матрице 100 x 100 x 20 000.

Язык

Я бы хотел использовать Tensorflow, потому что он позволит мне использовать либо python, либо C ++ ,

Есть ли другая библиотека, которая позволит мне ускорять это на GPU, например, Thrust или cuBLAS, которая будет работать?

Answer 1

Как насчет argsort ? Однако он не будет иметь такую ​​же обработку для связей.