fft

colibri.optics.functional.fft(field, axis=(-2, -1))

Fast Fourier Transform of an optical field

Parameters:

• field (torch.Tensor) – Input field.

• axis (tuple) – Tuple with the axes to perform the fft.

Returns:

Fourier transform of the input field.

Return type:

torch.Tensor