megengine.functional.nn.correlation

correlation(data1, data2, kernel_size=1, max_displacement=1, stride1=1, stride2=1, pad_size=0, is_multiply=True)[源代码]

计算两特征图之间的相关系数。

参数
  • data1 (Tensor) – 格式为 NCHW 的特征图1

  • data2 (Tensor) – 格式为 NCHW 的特征图2

  • kernel_size (int) – 非负奇整数,表示 patch 的大小,即计算时邻域大小。默认值为 1

  • max_displacement (int) – 非负整数,计算时的窗口大小。默认值为 1

  • stride1 (int) – 非负整数,特征图 1 上取值的步长。其值决定输出的宽和高。默认值为 1。

  • stride2 (int) – 非负整数,在特征图 2 中取特征图 1 的邻域的步长。默认值为1。

  • pad_size (int) – 非负整数,在四个方向分别对两输入补 0 的数量

  • is_multiply (bool) – 布尔值,设置为 True 时累积对应点的乘积,否则累积对应点差值的绝对值。默认为 True

返回类型

Tensor