Skip to content

Commit 45b7627

Browse files
authored
bugfix reshape -1 (#36143)
1 parent a57f081 commit 45b7627

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddle/fluid/operators/reshape_op.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -229,7 +229,7 @@ class ReshapeOp : public framework::OperatorWithKernel {
229229
// by now we require that if the input tensor is zero shape, the target
230230
// shape of output must be zero
231231
if (in_size == 0) {
232-
PADDLE_ENFORCE_EQ(
232+
PADDLE_ENFORCE_LE(
233233
capacity, in_size,
234234
platform::errors::InvalidArgument(
235235
"The 'shape' in ReshapeOp is invalid. "

0 commit comments

Comments
 (0)