torch.argsort(a, dim=1)
tensor([[2, 0, 3, 1],
[3, 2, 1, 0],
[2, 1, 0, 3],
[3, 2, 1, 0]])
dim=1 means row-wise argument sort
can u explain me how u got [2,0,3,1] for 1st row [ 0.0785, 1.5267, -0.8521, 0.4065] ???
bcoz we given index according to increasing order value in all other rows.
if u explain it i got my question answer too bro @Prashant_Kalikotay
@ptrblck @tancl @SimonW