Torch.fft.fft 2. () error reporting problem solution

Running someone else's source code also reports an error , So make sure it's not a code problem .

Asked someone , Should be fft Function corresponding torch Version of the problem ,torch1.8.0 After version

torch.fft.fft2

According to the online summary, I changed it

# Old edition            new edition 
torch.rfft   torch.fft.fft2
torch.irfft  torch.fft.ifft2

It's a mistake , The parameter definitions used in functions should also be different

It's not a version problem , notice pytorch There are only these two functions on the official website , So guess if I made a mistake

   terms of settlement , Put the source code torch.fft.fft2 Change to torch.fft.fftn That's all right.

def D(x, Dh_DFT, Dv_DFT):
    x_DFT = torch.fft.fftn(x, dim=(-2,-1)).cuda()
    Dh_x = torch.fft.ifftn(Dh_DFT*x_DFT, dim=(-2,-1)).real
    Dv_x = torch.fft.ifftn(Dv_DFT*x_DFT, dim=(-2,-1)).real
    return Dh_x, Dv_x

Then the code runs normally , To prove that the code really ran successfully , Because I don't know what happened recently , Start reporting the same mistakes as before , No reason found , But at least it worked , Then my change should be right .