dask.array.fft.fft_wrap

dask.array.fft.fft_wrap

dask.array.fft.fft_wrap(fft_func, kind=None, dtype=None, allow_fftpack=False)[source]

Wrap 1D, 2D, and ND real and complex FFT functions

Takes a function that behaves like numpy.fft functions and a specified kind to match it to that are named after the functions in the numpy.fft API.

Supported kinds include:

  • fft

  • fft2

  • fftn

  • ifft

  • ifft2

  • ifftn

  • rfft

  • rfft2

  • rfftn

  • irfft

  • irfft2

  • irfftn

  • hfft

  • ihfft

Examples

>>> import dask.array.fft as dff
>>> parallel_fft = dff.fft_wrap(np.fft.fft)
>>> parallel_ifft = dff.fft_wrap(np.fft.ifft)