coded_phase_imaging_forward

colibri.optics.functional.coded_phase_imaging_forward(field, phase_mask, distance, pixel_size, wavelength, approximation)[source]

Forward model for phase retrieval using a single distance and wavelength.

\[\mathbf{y} = \mathcal{P}_{(z, \lambda)}(\mathbf{x} \odot \learnedOptics)\]
Parameters:

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

  • distance (float) – Distance in meters.

  • wavelength (float) – Wavelength in meters.

  • field (Tensor)

  • phase_mask (Tensor)

  • pixel_size (float)

  • approximation (str)

Returns:

Phase retrieval measurement (B, 1, M, N).

Return type:

torch.Tensor