diff --git a/paddle/fluid/operators/save_op.cc b/paddle/fluid/operators/save_op.cc index f45d07ed90d52d204e9a3a5c2efe6df6b27ebfe6..dcc1b9ec204e9e273b8fd2b12f2423fc989ba502 100644 --- a/paddle/fluid/operators/save_op.cc +++ b/paddle/fluid/operators/save_op.cc @@ -106,6 +106,8 @@ class SaveOp : public framework::OperatorBase { auto out_kernel_type = framework::OpKernelType(out_dtype, place); framework::LoDTensor out; framework::TransDataType(in_kernel_type, out_kernel_type, tensor, &out); + // copy LoD info to the new tensor + out.set_lod(tensor.lod()); framework::SerializeToStream(fout, out, dev_ctx); } else { framework::SerializeToStream(fout, tensor, dev_ctx);