Thank you very much. the f is a rgb image, i want to compute its fft.
i just tried with FL = torch.rfft(f, 3) / torch.sqrt(M * N) .
it works well. i am a little puzzled. what doesthe parameter 3 mean for ?
rgb 3channels?