为了使用振幅和相位，我使用下面的函数来提取：

def calculate_fft(x):
    fft_im = torch.fft.fft(x.clone())  # bx3xhxw
    fft_amp = fft_im.real**2 + fft_im.imag**2
    fft_amp = torch.sqrt(fft_amp)
    fft_pha = torch.atan2(fft_im.imag, fft_im.real)
    return fft_amp, fft_pha

修改amp和pha后，如何使用它们执行逆FFT？

y = fft_amp * torch.sin(fft_pha)

英译汉这不起作用。我数学不好。