need to copy LoD info (#10392)

7 years ago · ccc594e4c4
parent 773bf29b70
commit ccc594e4c4
1 changed files with 2 additions and 0 deletions
--- 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);