numpy中有函數argsort來返回排序后的下標
pytorch中的torch.sort本身就能返回排序后的下標
Python中沒有直接調用的接口，怎么辦呢？
用enumerate再排序就可以了
nums = [4, 1, 5, 2, 9, 6, 8, 7] sorted_nums = sorted(enumerate(nums), key=lambda x: x[1]) idx = [i[0] for i in sorted_nums] nums = [i[1] for i in sorted_nums]