I am finding how could I do argsort with cuda/thrust along rows or colums of a matrix.
argsort
It means that given a matrix such as:
A = [[ 3.4257