cupyx.scipy.fft.fftn#

cupyx.scipy.fft.fftn(x, s=None, axes=None, norm=None, overwrite_x=False, *, plan=None)[source]#

计算 N 维 FFT。

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

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

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

  • norm ("backward", "ortho", 或 "forward") – 可选关键字,指定归一化模式。默认为 None,它是 "backward" 的别名。

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

  • plan (cupy.cuda.cufft.PlanNdNone) –

    一个 cuFFT 计划,用于在 axes 上变换 x,可以使用以下方法获得

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

    请注意,plan 默认为 None,这意味着 CuPy 会在后台使用自动生成的计划。

返回:

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

返回类型:

cupy.ndarray