add axis check for concat op (#21288)

* add axis check for concat op
test=develop

* fix PADDLE_ENFORCE format
test=develop

* move to ComputeAxis for InferShape check
test=develop

paddle/fluid/operators/concat_op.h

@@ -61,6 +61,11 @@ static inline framework::DDim ComputeAndCheckShape(
 }
 
 static inline int64_t ComputeAxis(int64_t axis, int64_t rank) {
+  PADDLE_ENFORCE_EQ(
+      axis >= -rank && axis < rank, true,
+      platform::errors::InvalidArgument(
+          "The axis is expected to be in range of [%d, %d), but got %d", -rank,
+          rank, axis));
   if (axis < 0) {
     axis = axis + rank;
   }