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

Question

问题：

SciPy有一个function，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 20000矩阵上的秩数据函数

语言

<>我想使用TysFooSurm，因为它将允许我使用Python或C++。p>

是否有一个不同的库，可以让我的GPU加速，像推力或库布拉斯，这将工作

Answer 1

那argsort呢？不过，领带的处理方式就不一样了