bugfix reshape -1 (#36087)

PaddlePaddle · Sep 26, 2021 · 2fe9ae7 · 2fe9ae7
1 parent e45d64e
commit 2fe9ae7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddle/fluid/operators/reshape_op.cc b/paddle/fluid/operators/reshape_op.cc
@@ -229,7 +229,7 @@ class ReshapeOp : public framework::OperatorWithKernel {
     // by now we require that if the input tensor is zero shape, the target
     // shape of output must be zero
     if (in_size == 0) {
-      PADDLE_ENFORCE_EQ(
+      PADDLE_ENFORCE_LE(
           capacity, in_size,
           platform::errors::InvalidArgument(
               "The 'shape' in ReshapeOp is invalid. "