方法一:使用Numpy的argsort函数
index = np.argsort(array)(array为待排序数据)
方法二:使用pytorch的sort函数
通过array_tensor=torch.Tensor(array)(array为待排序数据),将数据(如numpy的ndarray)转化为tensor数组,然后用value,index = torch.sort(array_tensor)即可得到排序后的value和index
方法一:使用Numpy的argsort函数
index = np.argsort(array)(array为待排序数据)
方法二:使用pytorch的sort函数
通过array_tensor=torch.Tensor(array)(array为待排序数据),将数据(如numpy的ndarray)转化为tensor数组,然后用value,index = torch.sort(array_tensor)即可得到排序后的value和index