conv_transpose2d_fft

class deepinv.physics.functional.conv_transpose2d_fft(y: Tensor, filter: Tensor, real_fft: bool = True)[source]

Bases:

A helper function performing the 2d transposed convolution 2d of x and filter using FFT. The adjoint of this operation is deepinv.physics.functional.conv2d_fft().

Parameters:
  • y (torch.Tensor) – Image of size (B, C, W, H).

  • filter (torch.Tensor) – Filter of size (b, c, w, h) ) where b can be either 1 or B and c can be either 1 or C.

If b = 1 or c = 1, then this function supports broadcasting as the same as numpy. Otherwise, each channel of each image is convolved with the corresponding kernel.

For convolution using FFT consider only 'circular' padding (i.e., circular convolution).

Returns:

torch.Tensor : the output of the convolution, which has the same shape as \(y\)