diff --git a/paddle/fluid/operators/dist_op.cc b/paddle/fluid/operators/dist_op.cc
index 0e9bddf01e82f82d15d2d4bbe481009898f7c414..0cb8c29d1145122fa75633915b889f60a59c288a 100644
--- a/paddle/fluid/operators/dist_op.cc
+++ b/paddle/fluid/operators/dist_op.cc
@@ -27,6 +27,20 @@ class DistOp : public framework::OperatorWithKernel {
     OP_INOUT_CHECK(ctx->HasInput("X"), "Input", "X", "Dist");
     OP_INOUT_CHECK(ctx->HasInput("Y"), "Input", "Y", "Dist");
     OP_INOUT_CHECK(ctx->HasOutput("Out"), "Output", "Out", "Dist");
+
+    auto x_dims = ctx->GetInputDim("X");
+    auto y_dims = ctx->GetInputDim("Y");
+
+    PADDLE_ENFORCE_NE(framework::product(x_dims), 0,
+                      platform::errors::InvalidArgument(
+                          "The Input(X) has not been initialized properly. The "
+                          "shape of Input(X) = [%s].",
+                          x_dims));
+    PADDLE_ENFORCE_NE(framework::product(y_dims), 0,
+                      platform::errors::InvalidArgument(
+                          "The Input(Y) has not been initialized properly. The "
+                          "shape of Input(Y) = [%s].",
+                          y_dims));
     ctx->SetOutputDim("Out", {1});
   }
 };