cupyx.scipy.signal.fftconvolve#

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

使用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 or tuple of scalar or None) – 计算卷积的轴。默认是所有轴。

返回: