torchcvnn.transforms.IFFT2

class torchcvnn.transforms.IFFT2(axis: Tuple[int, ...] = (-2, -1))[source]

Applies 2D inverse Fast Fourier Transform (IFFT) to the input. This transform computes the 2D IFFT along the last two dimensions of the input array/tensor. It applies inverse FFT shift before IFFT2.

Args
axisTuple[int, …], optional

The axes over which to compute the FFT. Default is (-2, -1).

Returns
numpy.ndarray or torch.Tensor:

The inverse Fourier transformed input. Output has the same shape as input.

Notes

  • Transform is applied along specified dimensions (axis).

__init__(axis: Tuple[int, ...] = (-2, -1))[source]

Methods

__init__([axis])