是否有类似于 SciPy 排名数据的 TF 函数

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做到这一点？我需要 GPU 加速 100 x 100 x 20,000 矩阵上的排名数据函数。

语言

我想使用Tensorflow，因为它可以让我使用python或C++。

有没有不同的库可以让我 GPU 加速它，比如 Thrust 或 cuBLAS，可以工作？

Answer 1

argsort怎么样？不过，它不会对领带有相同的处理方式。