提交 3ce12b1b 编写于 作者: C chengduozh

fix shape api doc

test=develop
上级 4c8feae4
......@@ -35,14 +35,15 @@ class ShapeOp : public framework::OperatorWithKernel {
class ShapeOpMaker : public framework::OpProtoAndCheckerMaker {
public:
void Make() override {
AddInput("Input", "(Tensor), The input tensor.");
AddOutput("Out",
"(Tensor), The shape of input tensor, the data type of the shape"
" is int32_t, will be on the same device with the input Tensor.");
AddInput("Input", "(LoDTensor), The input tensor.");
AddOutput(
"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");
}
};
......
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册