pado.math.sc_dft_2d

sc_dft_2d(u, Mx, My, delta_x, delta_y, delta_fx, delta_fy)[source]

Perform 2D scaled DFT using separable 1D transforms.

Parameters:

  • u (torch.Tensor) – Input field [My, Mx]

  • Mx (int) – Number of samples in x,y directions

  • My (int) – Number of samples in x,y directions

  • delta_x (float) – Spatial sampling intervals (m)

  • delta_y (float) – Spatial sampling intervals (m)

  • delta_fx (float) – Frequency sampling intervals (1/m)

  • delta_fy (float) – Frequency sampling intervals (1/m)

Returns:

Transformed field [My, Mx]

Return type:

torch.Tensor

Examples

>>> field = light.get_field().squeeze()
>>> U = sc_dft_2d(field, 1024, 1024, pitch, pitch, 1/(pitch*1024), 1/(pitch*1024))