未验证 提交 2fe9ae71 编写于 作者: J JZ-LIANG 提交者: GitHub

bugfix reshape -1 (#36087)

上级 e45d64ec
...@@ -229,7 +229,7 @@ class ReshapeOp : public framework::OperatorWithKernel { ...@@ -229,7 +229,7 @@ class ReshapeOp : public framework::OperatorWithKernel {
// by now we require that if the input tensor is zero shape, the target // by now we require that if the input tensor is zero shape, the target
// shape of output must be zero // shape of output must be zero
if (in_size == 0) { if (in_size == 0) {
PADDLE_ENFORCE_EQ( PADDLE_ENFORCE_LE(
capacity, in_size, capacity, in_size,
platform::errors::InvalidArgument( platform::errors::InvalidArgument(
"The 'shape' in ReshapeOp is invalid. " "The 'shape' in ReshapeOp is invalid. "
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册