Unable to backpropagate through torch.rfft

Question

I have two losses: one the usual L1 loss and second one involving torch.rfft()

def dft_amp(img):
    fft_im = torch.rfft( img, signal_ndim=2, oneside