cupyx.scipy.fftpack.rfft#

cupyx.scipy.fftpack.rfft(x, n=None, axis=-1, overwrite_x=False, plan=None)[源代码][源代码]#

计算实输入的一维快速傅里叶变换。

返回的实数数组包含

[y(0),Re(y(1)),Im(y(1)),...,Re(y(n/2))]  # if n is even
[y(0),Re(y(1)),Im(y(1)),...,Re(y(n/2)),Im(y(n/2))]  # if n is odd
参数:
  • x (cupy.ndarray) – 要转换的数组。

  • n (None or int) – 输出转换轴的长度。如果未给出 n ,则使用沿 axis 指定的输入轴的长度。

  • axis (int) – 计算FFT的轴。

  • overwrite_x (bool) – 如果为真,x 的内容可以被销毁。

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

    用于在 axis 上变换 x 的 cuFFT 计划,可以通过以下方式获得:

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

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

返回:

转换后的数组。

返回类型:

cupy.ndarray

备注

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