cupyx.scipy.fft.rfft2#

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

计算实数输入的二维 FFT。

参数:

a (cupy.ndarray) – 要进行变换的数组。
s (None 或 整数元组) – 从输入中使用的形状。如果未给出 s，则使用输入沿由 axes 指定的轴的长度。
axes (整数元组) – 计算 FFT 的轴。
norm ("backward", "ortho", 或 "forward") – 可选关键字，用于指定归一化模式。默认为 None，它是 "backward" 的别名。
overwrite_x (bool) – 如果为 True，则 x 的内容可能会被销毁。
plan (cupy.cuda.cufft.PlanNd 或 None) –
用于对 axes 上的 x 进行变换的 cuFFT 计划，可以通过以下方式获取
```
plan = cupyx.scipy.fftpack.get_fft_plan(x, s, axes,
                                        value_type='R2C')
```
请注意，plan 默认为 None，这意味着 CuPy 将在后台使用自动生成的计划。

返回:

变换后的数组，其形状由 s 指定，如果输入为其他类型，则类型将转换为复数。最后一个变换轴的长度将为 s[-1]//2+1。

返回类型:

cupy.ndarray

另请参阅