update comment

PaddlePaddle · Sep 7, 2020 · 7ce82e1 · 7ce82e1
1 parent 0b429e8
commit 7ce82e1
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/paddle/fluid/operators/correlation_op.cc b/paddle/fluid/operators/correlation_op.cc
@@ -52,7 +52,9 @@ class CorrelationOpMaker : public framework::OpProtoAndCheckerMaker {
   void Make() override {
     AddInput("Input1", "Input is a 4-D Tensor with shape [N, C, H, W]");
     AddInput("Input2", "Input is a 4-D Tensor with shape [N, C, H, W]");
-    AddOutput("Output", "Output of CorrelationOp");
+    AddOutput("Output",
+              "(Tensor) The output tensor of correlation operator. "
+              "It has same data fromat and data type as the Input.");
     AddAttr<int>("pad_size", "pad size for input1 and input2");
     AddAttr<int>("kernel_size", "kernel size of input1 and input2");
     AddAttr<int>("max_displacement", "max displacement of input1 and input2");