diff --git a/paddle/fluid/operators/assign_op.h b/paddle/fluid/operators/assign_op.h index e90de55bbbd9035b495b067046fed162b36e5f0d..e6374eb3a66f72283de0845a170f4956b784fa4d 100644 --- a/paddle/fluid/operators/assign_op.h +++ b/paddle/fluid/operators/assign_op.h @@ -71,7 +71,7 @@ class AssignFunctor { private: void copy_tensor(const framework::LoDTensor &lod_tensor, framework::LoDTensor *out) const { - if (lod_tensor.numel() == 0) return; + if (!lod_tensor.IsInitialized()) return; auto &out_tensor = *out; paddle::framework::TensorCopy(lod_tensor, lod_tensor.place(), &out_tensor); out_tensor.set_lod(lod_tensor.lod());