FFT

class complextorch.nn.modules.fft.FFTBlock(n=None, dim=-1, norm=None)

FFT Block

A complex-valued module that performs the forward fast Fourier transform.

For more information, see PyTorch fft <https://pytorch.org/docs/stable/fft.html>.

forward(input: CVTensor) → CVTensor

Performs forward fast Fourier transform (FFT) on the input tensor.

Parameters:

input (CVTensor) – input tensor

Returns:

fft(x) using configuration established on initialization

Return type:

CVTensor

class complextorch.nn.modules.fft.IFFTBlock(n=None, dim=-1, norm=None)

IFFT Block

A complex-valued module that performs the inverse fast Fourier transform.

For more information, see PyTorch fft <https://pytorch.org/docs/stable/fft.html>.

forward(input: CVTensor) → CVTensor

Performs inverse fast Fourier transform (FFT) on the input tensor.

Parameters:

input (CVTensor) – input tensor

Returns:

ifft(x) using configuration established on initialization

Return type:

CVTensor