提交 560b29cc 编写于 作者: M minqiyang

Polish code

test=develop
上级 21d6e8e8
......@@ -117,7 +117,7 @@ class ExpandGradOp : public framework::OperatorWithKernel {
size_t start_pos = 0u;
if (!ctx->IsRuntime()) {
PADDLE_ENFORCE_EQ(
x_dims[i], out_dims[i],
x_dims[0], out_dims[0],
"The first dimension size of Input(Out@GRAD) should be "
"equal to the crroresponding dimension size of Input(X)");
start_pos = 1u;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册