提交 0ea46822 编写于 作者: C chenweihang

docs: fix some errors of description

上级 2bd65dbf
...@@ -73,7 +73,7 @@ class SqueezeOpInferShape : public framework::InferShapeBase { ...@@ -73,7 +73,7 @@ class SqueezeOpInferShape : public framework::InferShapeBase {
"Invalid axis, the negative axis is out of range."); "Invalid axis, the negative axis is out of range.");
PADDLE_ENFORCE(in_dims[current] == 1, PADDLE_ENFORCE(in_dims[current] == 1,
"Invalid axis index, the axis that will be squeezed " "Invalid axis index, the axis that will be squeezed "
"should equal 1."); "should be equal to 1.");
if (!(should_squeeze[current])) { if (!(should_squeeze[current])) {
++cnt_squeezed_dims; ++cnt_squeezed_dims;
...@@ -123,7 +123,7 @@ class SqueezeOpMaker : public framework::OpProtoAndCheckerMaker { ...@@ -123,7 +123,7 @@ class SqueezeOpMaker : public framework::OpProtoAndCheckerMaker {
AddOutput("Out", "(Tensor). The output tensor of squeeze operator."); AddOutput("Out", "(Tensor). The output tensor of squeeze operator.");
AddAttr<std::vector<int>>("axes", AddAttr<std::vector<int>>("axes",
"(std::vector<int>). List of integers," "(std::vector<int>). List of integers,"
" indicate the dimensions to squeeze.") " indicating the dimensions to squeeze.")
.SetDefault({}); .SetDefault({});
AddAttr<bool>("inplace", AddAttr<bool>("inplace",
"(default: false) Squeeze the source tensor's shape without " "(default: false) Squeeze the source tensor's shape without "
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册