fourier_conv

colibri.optics.functional.fourier_conv(image, psf)

Apply Fourier convolution theorem to simulate the effect of a linear system characterized by a point spread function (PSF).

\[g = \mathcal{F}^{-1}(\mathcal{F}(f) * \mathcal{F}(h))\]

where \(f\) is the input image, \(h\) is the PSF, \(g\) is the convolved output, \(\mathcal{F}\) and \(\mathcal{F}^{-1}\) denote the Fourier and inverse Fourier transforms.

Parameters:

• image (torch.Tensor) – Image to simulate the sensing (B, L, M, N)

• psf (torch.Tensor) – Point Spread Function (1, L, M, N)

Returns:

Measurement (B, 1, M, N)

Return type:

torch.Tensor