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

docs: fix some errors of description

上级 2bd65dbf
......@@ -73,7 +73,7 @@ class SqueezeOpInferShape : public framework::InferShapeBase {
"Invalid axis, the negative axis is out of range.");
PADDLE_ENFORCE(in_dims[current] == 1,
"Invalid axis index, the axis that will be squeezed "
"should equal 1.");
"should be equal to 1.");
if (!(should_squeeze[current])) {
++cnt_squeezed_dims;
......@@ -123,7 +123,7 @@ class SqueezeOpMaker : public framework::OpProtoAndCheckerMaker {
AddOutput("Out", "(Tensor). The output tensor of squeeze operator.");
AddAttr<std::vector<int>>("axes",
"(std::vector<int>). List of integers,"
" indicate the dimensions to squeeze.")
" indicating the dimensions to squeeze.")
.SetDefault({});
AddAttr<bool>("inplace",
"(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.
先完成此消息的编辑!
想要评论请 注册