def psnr(img1, img2): * mse = (((img1 - img2)) ** 2).view(img1.shape[0], -1).mean(1, keepdim=True) * return 20 * torch.log10(1.0 / torch.sqr

sorts vectors from large to small * vec: vector to be sorted * sorted_vec: sorted results * ind: the position of each element in the sort result in