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: 使用 fillvalue 填充输入数组(默认)

    • wrap: 循环边界条件

    • symm: 对称边界条件

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

返回:

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

返回类型:

cupy.ndarray