cupyx.scipy.signal.convolve2d#

cupyx.scipy.signal.convolve2d(in1, in2, mode='full', boundary='fill', fillvalue=0)[源代码][源代码]#

卷积两个二维数组。

in1in2 进行卷积,输出大小由 mode 决定,边界条件由 boundaryfillvalue 决定。

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

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

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

  • boundary (str) – 指示如何处理边界: - fill:用填充值填充输入数组(默认) - wrap:循环边界条件 - symm:对称边界条件

  • fillvalue (scalar) – 用于填充输入数组的值。默认值为 0。

返回:

一个二维数组,包含 in1in2 的离散线性卷积的子集。

返回类型:

cupy.ndarray