test=develop

paddle/fluid/operators/sequence_ops/sequence_mask_op.h

@@ -36,12 +36,10 @@ class SequenceMaskOp : public framework::OperatorWithKernel {
     PADDLE_ENFORCE(ctx->HasInput("X"), "Input(X) must exist");
     PADDLE_ENFORCE(ctx->HasOutput("Y"), "Output(Y) must exist");
 
-    auto maxlen = ctx->Attrs().Get<int>("maxlen");
+    int maxlen = ctx->Attrs().Get<int>("maxlen");
-    if (maxlen > 0) {  // We can only infershape when maxlen > 0
+    auto dim = framework::vectorize2int(ctx->GetInputDim("X"));
-      auto dim = framework::vectorize2int(ctx->GetInputDim("X"));
+    dim.push_back(maxlen > 0 ? maxlen : -1);
-      dim.push_back(maxlen);
+    ctx->SetOutputDim("Y", framework::make_ddim(dim));
-      ctx->SetOutputDim("Y", framework::make_ddim(dim));
-    }
   }
 };