diff --git a/paddle/fluid/framework/operator.cc b/paddle/fluid/framework/operator.cc
index 72dec87847b7be52c6b582eb9bf1859df6adcd2e..9baf720429d3a42f852fbd995637d9ef5a3984e7 100644
--- a/paddle/fluid/framework/operator.cc
+++ b/paddle/fluid/framework/operator.cc
@@ -1086,7 +1086,9 @@ void OperatorWithKernel::TransferInplaceVarsBack(
     PADDLE_ENFORCE_NOT_NULL(var, "The var[%s] should not be nullptr.",
                             var_name);
     auto* transformed_tensor = GetLoDTensorOrSelectedRowsValueFromVar(*var);
+    auto original_dims = original_tensor->dims();
     original_tensor->ShareDataWith(*transformed_tensor);
+    original_tensor->Resize(original_dims);
   }
 }