Merge pull request #16259 from phlrain/fix_lod_reset

Fix lod reset
6 years ago · b3b7e352ee
parent da39a70416 9874530ee8
commit b3b7e352ee
1 changed files with 3 additions and 0 deletions
--- a/paddle/fluid/operators/lod_reset_op.cc
+++ b/paddle/fluid/operators/lod_reset_op.cc
@ -32,7 +32,10 @@ class LoDResetOp : public framework::OperatorWithKernel {
      PADDLE_ENFORCE_GT(level0.size(), 1,
                        "If Input(Y) not provided, the target lod should be "
                        "specified by attribute `target_lod`.");
+    } else {
+      ctx->ShareLoD("Y", "Out");
    }
+
    ctx->SetOutputDim("Out", ctx->GetInputDim("X"));
  }