A similar method exists in NumPy (https://docs.scipy.org/doc/numpy/reference/generated/numpy.argsort.html) and in TensorFlow (https://www.tensorflow.org/api_docs/python/tf/contrib/framework/argsort).

Something like:

def argsort(x, *args, **kwargs);
   return x.sort(*args, **kwargs)[1]