fix concat dimension (#25927)

Fix the condition of concat dimension judgment.

fix concat dimension (#25927)
Fix the condition of concat dimension judgment.
05603c9a · GaoWei8 · GitHub · 7c111a08 · 05603c9a
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddle/fluid/operators/concat_op.h paddle/fluid/operators/concat_op.h +1 -1

未找到文件。
--- 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],