Skip to content

Commit 560b29c

Browse files
committed
Polish code
test=develop
1 parent 21d6e8e commit 560b29c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddle/fluid/operators/expand_op.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -117,7 +117,7 @@ class ExpandGradOp : public framework::OperatorWithKernel {
117117
size_t start_pos = 0u;
118118
if (!ctx->IsRuntime()) {
119119
PADDLE_ENFORCE_EQ(
120-
x_dims[i], out_dims[i],
120+
x_dims[0], out_dims[0],
121121
"The first dimension size of Input(Out@GRAD) should be "
122122
"equal to the crroresponding dimension size of Input(X)");
123123
start_pos = 1u;

0 commit comments

Comments
 (0)