dft_kernel

ska_sdp_func_python.imaging.dft.dft_kernel(direction_cosines, vfluxes, uvw_lambda, dft_compute_kernel=None)[source]

CPU computational kernel for DFT, choice dependent on dft_compute_kernel.

Parameters:

  • direction_cosines – Direction cosines [ncomp, 3]

  • vfluxes – Fluxes [ncomp, nchan, npol]

  • uvw_lambda – UVW in lambda [ntimes, nbaselines, nchan, 3]

  • dft_compute_kernel – string: cpu_looped, gpu_cupy_raw or proc_func

Returns:

Vis array, dims: [ntimes, nbaselines, nchan, npol]