dask.array.fft.fft_wrap

dask.array.fft.fft_wrap

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

封装一维、二维和多维实数和复数FFT函数

接受一个行为类似于 numpy.fft 函数的函数,并指定一种类型以匹配 numpy.fft API 中命名的函数。

支持的类型包括:

  • fft

  • fft2

  • fftn

  • ifft

  • ifft2

  • ifftn

  • rfft

  • rfft2

  • rfftn

  • irfft

  • irfft2

  • irfftn

  • hfft

  • ihfft

示例

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