The outputs of torch.fft.irfft and torch.fft.irfftn are real. Would it be possible, by any chance, to the the complex version of it, i.e., before the result is clipped to the real domain?
I already went through these code on github and I could not find a way to, yet.

ATen/native/SpectralOps
ATen/native/cuda/SpectralOps