Polish code

velconia · velconia · commit 21d6e8e8c80d · 2018-11-15T19:52:40.000+08:00
test=develop
diff --git a/paddle/fluid/operators/expand_op.cc b/paddle/fluid/operators/expand_op.cc
@@ -116,6 +116,10 @@ class ExpandGradOp : public framework::OperatorWithKernel {
 
     size_t start_pos = 0u;
     if (!ctx->IsRuntime()) {
+      PADDLE_ENFORCE_EQ(
+          x_dims[i], out_dims[i],
+          "The first dimension size of Input(Out@GRAD) should be "
+          "equal to the crroresponding dimension size of Input(X)");
       start_pos = 1u;
     }
 

Original file line number	Diff line number	Diff line change
`@@ -116,6 +116,10 @@ class ExpandGradOp : public framework::OperatorWithKernel {`
`116`	`116`
`117`	`117`	`size_t start_pos = 0u;`
`118`	`118`	`if (!ctx->IsRuntime()) {`
	`119`	`+ PADDLE_ENFORCE_EQ(`
	`120`	`+ x_dims[i], out_dims[i],`
	`121`	`+ "The first dimension size of Input(Out@GRAD) should be "`
	`122`	`+ "equal to the crroresponding dimension size of Input(X)");`
`119`	`123`	`start_pos = 1u;`
`120`	`124`	`}`
`121`	`125`