From 560b29ccb7cced8ed0756fd9b3f6dce9dc771483 Mon Sep 17 00:00:00 2001
From: minqiyang <minqiyang@baidu.com>
Date: Thu, 15 Nov 2018 19:53:55 +0800
Subject: [PATCH] Polish code

test=develop
---
 paddle/fluid/operators/expand_op.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/paddle/fluid/operators/expand_op.cc b/paddle/fluid/operators/expand_op.cc
index eef23830221..40f7c1c54c8 100644
--- a/paddle/fluid/operators/expand_op.cc
+++ b/paddle/fluid/operators/expand_op.cc
@@ -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;
-- 
GitLab