Merge pull request #16357 from phlrain/fix_concat_check

Fix concat check
6 years ago · e3bca9011c
parent e5478ab5c8 ff112813de
commit e3bca9011c
1 changed files with 1 additions and 1 deletions
--- a/paddle/fluid/operators/concat_op.cc
+++ b/paddle/fluid/operators/concat_op.cc
@ -57,7 +57,7 @@ class ConcatOp : public framework::OperatorWithKernel {
                              "elements except the specify axis.");
          } else {
            // not check -1 with other in compile time
-            if (out_dims[j] != -1 && ins[i][j] != -1) {
+            if (out_dims[j] > 0 && ins[i][j] > 0) {
              PADDLE_ENFORCE_EQ(out_dims[j], ins[i][j],
                                "Input tensors should have the same "
                                "elements except the specify axis.");