cupyx.scipy.signal.correlate2d#

cupyx.scipy.signal.correlate2d(in1, in2, mode='full', boundary='fill', fillvalue=0)[source]#

对两个二维数组进行互相关操作。

根据 mode 确定输出大小,并根据 boundaryfillvalue 确定边界条件,对 in1in2 进行互相关。

参数:
  • 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

注意

当使用 "same" 模式处理偶数长度输入时,correlatecorrelate2d 的输出不同:它们之间存在 1 个索引的偏移。