cupyx.scipy.fftpack.fftn#

cupyx.scipy.fftpack.fftn(x, shape=None, axes=None, overwrite_x=False, plan=None)[源代码]#

计算 N 维 FFT。

参数：

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

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

  • axes (整数元组) – 计算 FFT 的轴。

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

  • plan (cupy.cuda.cufft.PlanNd 或 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.fftn()

注意

参数 plan 目前是实验性的，其接口在未来版本中可能会改变。