I am looking for numpy.lexsort equivalent function in pytorch. Thanks.
Example:

import numpy as  np


a = np.array([[18.    25.     3.559  0.   ]
[18.    25.     3.559  0.   ]
[18.    25.     3.559  0.   ]
[18.    25.     3.56   0.09 ]
[15.    25.     3.447  0.2  ]
[14.    25.     3.445  0.58 ]
[15.    25.     3.453  0.   ]
[18.    26.     3.559  0.   ]
[18.    26.     3.558  0.   ]
[18.    26.     3.558  0.   ]])


indices = np.lexsort((a[:,2], a[:,1], a[:, 0]))
b=a[indices]


print(b)

Output:

[[14.    25.     3.445  0.58 ]
[15.    25.     3.447  0.2  ]
[15.    25.     3.453  0.   ]
[18.    25.     3.559  0.   ]
[18.    25.     3.559  0.   ]
[18.    25.     3.559  0.   ]
[18.    25.     3.56   0.09 ]
[18.    26.     3.558  0.   ]
[18.    26.     3.558  0.   ]
[18.    26.     3.559  0.   ]]