cupyx.scipy.fftpack.fft2#

cupyx.scipy.fftpack.fft2(x, shape=None, axes=(-2, -1), overwrite_x=False, plan=None)[source]#

计算二维 FFT。

参数:

  • x (cupy.ndarray) – 要进行变换的数组。

  • shape (None 或 tuple of ints) – 输出数组变换轴的形状。如果未指定 shape，则使用输入数组沿 axes 指定的轴的长度。

  • axes (tuple of ints) – 计算 FFT 的轴。

  • overwrite_x (bool) – 如果为 True，则 x 的内容可能会被销毁。

  • plan (cupy.cuda.cufft.PlanNd or None) –

    用于对 axes 上的 x 进行变换的 cuFFT 计划，可以通过以下方式获得。

    plan = cupyx.scipy.fftpack.get_fft_plan(x, axes)

    请注意，plan 默认为 None，这意味着如果 cupy.fft.config. enable_nd_planning = True，CuPy 将在内部使用自动生成的计划；如果设置为 False，则不使用 cuFFT 计划。

返回:

变换后的数组，其形状由 shape 指定，如果输入类型不是复数类型，则输出类型将转换为复数类型。

返回类型:

cupy.ndarray

另请参阅

scipy.fftpack.fft2()

注意

参数 plan 目前处于实验阶段，其接口在未来版本中可能会发生变化。