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

bugfix reshape -1 (#36143)

上级 a57f0810
......@@ -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. "
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册