diff --git a/paddle/fluid/inference/api/analysis_predictor.cc b/paddle/fluid/inference/api/analysis_predictor.cc
index 0190d9c291b8ecc0697a7584913994ed40a83ca1..8663ec7d1f09be59888de0b39048a2d1eadfc669 100644
--- a/paddle/fluid/inference/api/analysis_predictor.cc
+++ b/paddle/fluid/inference/api/analysis_predictor.cc
@@ -182,6 +182,11 @@ bool PaddleTensorToLoDTensor(const PaddleTensor &pt,
         pt.data.data(),
         paddle::platform::errors::InvalidArgument(
             "The data contained in the input PaddleTensor is illegal."));
+    PADDLE_ENFORCE_EQ(
+        pt.data.length(),
+        t->numel() * paddle::experimental::SizeOf(t->dtype()),
+        paddle::platform::errors::InvalidArgument(
+            "The data contained in the input PaddleTensor had wrong length."));
   }
 
   if (platform::is_cpu_place(place)) {
diff --git a/paddle/fluid/inference/api/api_impl.cc b/paddle/fluid/inference/api/api_impl.cc
index 865c08fe2f64def5c8790fbfe4eea49f995bd10c..1b6779ac9fbc91920baa15e32b0875c43f4eef72 100644
--- a/paddle/fluid/inference/api/api_impl.cc
+++ b/paddle/fluid/inference/api/api_impl.cc
@@ -234,6 +234,12 @@ bool NativePaddlePredictor::SetFeed(const std::vector<PaddleTensor> &inputs,
         inputs[i].data.data(),
         platform::errors::InvalidArgument(
             "The data of input tensor should not be null."));
+    PADDLE_ENFORCE_EQ(
+        inputs[i].data.length(),
+        input.numel() * paddle::experimental::SizeOf(input.dtype()),
+        paddle::platform::errors::InvalidArgument(
+            "The data contained in the input PaddleTensor had wrong length."));
+
     if (platform::is_cpu_place(place_)) {
       // TODO(panyx0718): Init LoDTensor from existing memcpy to save a copy.
       std::memcpy(static_cast<void *>(input_ptr),
diff --git a/paddle/fluid/inference/tests/api/analyzer_transformer_tester_helper.h b/paddle/fluid/inference/tests/api/analyzer_transformer_tester_helper.h
index 68167c2a313a91839574e5a75f711c139b6941d8..569d62f637ff143a93cf490f4b41fa9025c8d929 100644
--- a/paddle/fluid/inference/tests/api/analyzer_transformer_tester_helper.h
+++ b/paddle/fluid/inference/tests/api/analyzer_transformer_tester_helper.h
@@ -154,7 +154,7 @@ void PrepareInputs(std::vector<PaddleTensor> *input_slots,
 
   init_idx.name = "init_idx";
   init_idx.shape.assign({batch_size});
-  init_idx.dtype = PaddleDType::INT32;
+  init_idx.dtype = PaddleDType::INT64;
   TensorAssignData<int64_t>(&init_idx, one_batch.init_idx);
 
   trg_src_attn_bias.name = "trg_src_attn_bias";