description: TensorFlow variant of NumPy's argsort.

tf.experimental.numpy.argsort

View source on GitHub

TensorFlow variant of NumPy's argsort.

tf.experimental.numpy.argsort(
    a, axis=-1, kind='quicksort', order=None
)

See the NumPy documentation for numpy.argsort.