diff --git a/paddle/fluid/operators/sequence_ops/sequence_pool_op.cc b/paddle/fluid/operators/sequence_ops/sequence_pool_op.cc index 1ece8bf937..09dba54028 100644 --- a/paddle/fluid/operators/sequence_ops/sequence_pool_op.cc +++ b/paddle/fluid/operators/sequence_ops/sequence_pool_op.cc @@ -31,9 +31,11 @@ class SequencePoolOp : public framework::OperatorWithKernel { if (!ctx->IsRuntime()) { // Check the lod_level for compile-time. + auto in_lod_level = ctx->GetLoDLevel("X"); PADDLE_ENFORCE_GT( - ctx->GetLoDLevel("X"), 0, + in_lod_level, 0, "The LoD level Input(X) of sequence_pool should be larger than 0."); + ctx->SetLoDLevel("Out", in_lod_level - 1); } ctx->SetOutputDim("Out", ctx->GetInputDim("X"));