Update common_shape.h

PaddlePaddle · Apr 9, 2024 · 9fc3081 · 9fc3081
1 parent 009cf48
commit 9fc3081
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/paddle/phi/kernels/funcs/common_shape.h b/paddle/phi/kernels/funcs/common_shape.h
@@ -309,8 +309,8 @@ inline std::vector<int64_t> GetReduceDims(const DenseTensor &in,
       reduce_dims.push_back(i + diff);
     } else {
       PADDLE_ENFORCE_EQ(
-          in_dims[i],
-          out_dims[i + diff],
+          in_dims[i + diff],
+          out_dims[i],
           phi::errors::InvalidArgument(
               "ReduceDims dimension mismatch. Operands could "
               "not be broadcast together with the shape of in_dims = [%s] and "