Is there a way to perform this operation more efficiently?
   sort1 = torch.sort(abs(u.grad))[1]
   srtcoe= u.grad.clone()
   for li in range(z.lrsize[lr]):
      for pi in range(z.lrsize[lr-1]):
         srtcoe[li][sort1[li][pi]] = pi /z.lrsize[pr]