cupyx.scipy.signal.oaconvolve#

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

使用重叠相加法卷积两个N维数组。

使用重叠相加法对 in1in2 进行卷积,输出大小由 mode 参数决定。对于大型数组,这通常比 convolve 更快,当一个数组远大于另一个数组时,通常也比 fftconvolve 更快,但在只需要几个输出值或数组形状非常相似时可能会更慢,并且只能输出浮点数组(整数或对象数组输入将被转换为浮点数)。

参数:
  • in1 (cupy.ndarray) – 第一个输入。

  • in2 (cupy.ndarray) – 第二个输入。应与 in1 具有相同的维度。

  • mode (str) – 指示输出的大小: - 'full':输出是完整的离散线性互相关(默认) - 'valid':输出仅包含那些不依赖于零填充的元素。in1in2 在每个维度上必须至少与另一个一样大。 - 'same':输出与 in1 大小相同,相对于 'full' 输出居中。

  • axes (scalar or tuple of scalar or None) – 计算卷积的轴。默认是所有轴。

返回:

卷积的结果

返回类型:

cupy.ndarray