diff --git a/paddle/fluid/operators/metrics/auc_op.cc b/paddle/fluid/operators/metrics/auc_op.cc
index 4f2f1d0722c9ce4bb47dd043da5489d54ec4884a..4ab101a627e3c421119e92e44ff0f6c0205b4fc8 100644
--- a/paddle/fluid/operators/metrics/auc_op.cc
+++ b/paddle/fluid/operators/metrics/auc_op.cc
@@ -27,6 +27,13 @@ class AucOp : public framework::OperatorWithKernel {
     OP_INOUT_CHECK(ctx->HasInput("Label"), "Input", "Label", "Auc");
     auto predict_dims = ctx->GetInputDim("Predict");
     auto label_dims = ctx->GetInputDim("Label");
+    PADDLE_ENFORCE_GE(
+        predict_dims.size(), 2,
+        platform::errors::InvalidArgument(
+            "The Input(Predict) has not been initialized properly. The "
+            "shape of Input(Predict) = [%s], the shape size must be "
+            "greater_equal 2.",
+            predict_dims));
     auto predict_width = predict_dims[1];
     PADDLE_ENFORCE_NE(
         framework::product(predict_dims), 0,