Как вычислить ранги массива в Tensorflow?

Question

Как получить рейтинг массива, например pandas .DataFrame.rank ()?

Например, для этого массива:

a = tf.constant([0, 2, 3, 3])

Результат, которого я ожидал это:

([0, 1, 2, 2])

Answer 1

Вот ответ, который я нахожу:

__, rank = tf.unique(a)
print(rank)
<tf.Tensor: shape=(4,), dtype=int32, numpy=array([0, 1, 2, 2], dtype=int32)>