cupyx.scipy.fft.fft2#

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

计算二维 FFT。

参数:

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

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

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

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

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

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

    一个用于在 axes 上变换 x 的 cuFFT 计划，可通过以下方式获取

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

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

返回值:

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

返回类型:

cupy.ndarray

另请参阅

scipy.fft.fft2()