cupyx.scipy.signal.fftconvolve#

cupyx.scipy.signal.fftconvolve(in1, in2, mode='full', axes=None)[source]#

使用 FFT 对两个 N 维数组进行卷积。

使用快速傅里叶变换方法对 in1 和 in2 进行卷积，输出大小由 mode 参数确定。

对于大型数组，这通常比 convolve 的 'direct' 方法快得多，但在只需要少数输出值时可能较慢，并且只能输出浮点数组（整数或对象数组输入将被转换为浮点数）。

参数:

in1 (cupy.ndarray) – 第一个输入。
in2 (cupy.ndarray) – 第二个输入。应与 in1 具有相同的维数。
mode (str) –
指示输出的大小
- 'full'：输出是完整的离散线性互相关（默认）
- 'valid'：输出仅包含不依赖于零填充的元素。`in1` 或 `in2` 在每个维度上必须至少与另一个一样大。
- 'same'：输出大小与 `in1` 相同，相对于 'full' 输出居中
axes (scalar 或 tuple of scalar 或 None) – 计算卷积所沿的轴。默认是在所有轴上计算。

返回: