fix shape api doc

test=develop

paddle/fluid/operators/shape_op.cc

@@ -35,14 +35,15 @@ class ShapeOp : public framework::OperatorWithKernel {
 class ShapeOpMaker : public framework::OpProtoAndCheckerMaker {
  public:
   void Make() override {
-    AddInput("Input", "(Tensor), The input tensor.");
+    AddInput("Input", "(LoDTensor), The input tensor.");
-    AddOutput("Out",
+    AddOutput(
-              "(Tensor), The shape of input tensor, the data type of the shape"
+        "Out",
+        "(LoDTensor), The shape of input tensor, the data type of the shape"
         " is int32_t, will be on the same device with the input Tensor.");
     AddComment(R"DOC(
-Shape Operator
+Shape Operator.
 
-Get the shape of input tensor. Only support CPU input Tensor now.
+Return the shape of the input.
 )DOC");
   }
 };

python/paddle/fluid/layers/nn.py

@@ -8710,13 +8710,15 @@ def slice(input, axes, starts, ends):
 @templatedoc()
 def shape(input):
     """
-    ${comment}
+    **Shape Layer**
+
+    Return the shape of the input.
 
     Args:
-        input (Variable): ${input_comment}
+        input (Variable): The input variable.
 
     Returns:
-        out (Variable): ${out_comment}
+        out (Variable): The shape of the input variable.
 
     Examples:
         .. code-block:: python