torch.argsort() function
b=torch.tensor([[10,5,78],
[1,108,98],
[90,72,2]])
b.argsort(axis=0)
when i m performing this, i have not got correct output:
my output is :
[[1, 0, 2],
[0, 2, 0],
[2, 1, 1]]
But correct output is  after sorting is to this is below, but i haven’t got this output :
[[1, 0, 1],
[0, 2, 2],
[2, 1, 0]]