fix concat dimension (#25927)

GaoWei8 · web-flow · commit 05603c9aa1b9 · 2020-08-04T10:09:49.000+08:00
Fix the condition of concat dimension judgment.
diff --git a/paddle/fluid/operators/concat_op.h b/paddle/fluid/operators/concat_op.h
@@ -51,7 +51,7 @@ static inline framework::DDim ComputeAndCheckShape(
         }
       } else {
         bool check_shape =
-            is_runtime || (out_dims[j] > 0 && inputs_dims[i][j] > 0);
+            is_runtime || (inputs_dims[0][j] > 0 && inputs_dims[i][j] > 0);
         if (check_shape) {
           // check all shape in run time
           PADDLE_ENFORCE_EQ(inputs_dims[0][j], inputs_dims[i][j],

Original file line number	Diff line number	Diff line change
`@@ -51,7 +51,7 @@ static inline framework::DDim ComputeAndCheckShape(`
`51`	`51`	`}`
`52`	`52`	`} else {`
`53`	`53`	`bool check_shape =`
`54`		`- is_runtime \|\| (out_dims[j] > 0 && inputs_dims[i][j] > 0);`
	`54`	`+ is_runtime \|\| (inputs_dims[0][j] > 0 && inputs_dims[i][j] > 0);`
`55`	`55`	`if (check_shape) {`
`56`	`56`	`// check all shape in run time`
`57`	`57`	`PADDLE_ENFORCE_EQ(inputs_dims[0][j], inputs_dims[i][j],`