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 thenumpy.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)