Function correlation

CV/PWCNet/correlation_op/correlation.py:21–25

(input1, input2, pad_size, kernel_size, max_displacement, stride1, stride2, corr_type_multiply=1)

from paddle.fluid.layer_helper import LayerHelper

def correlation(input1, input2, pad_size, kernel_size, max_displacement, stride1, stride2, corr_type_multiply=1):
  helper = LayerHelper("correlation", **locals())
  output = helper.create_variable_for_type_inference(dtype=input1.dtype)
  helper.append_op(type="correlation", inputs={"Input1": input1, "Input2": input2}, attrs={"pad_size": pad_size, "kernel_size": kernel_size, "max_displacement": max_displacement, "stride1": stride1, "stride2": stride2, "corr_type_multiply": corr_type_multiply}, outputs = {"Output": output})
  return output