diff --git a/paddle/fluid/operators/expand_op.cc b/paddle/fluid/operators/expand_op.cc index 40f7c1c54c861abebc84428f55e2769ac8969f0f..d3cf0947959a45c371ebc831b3600c71f516bd74 100644 --- a/paddle/fluid/operators/expand_op.cc +++ b/paddle/fluid/operators/expand_op.cc @@ -48,7 +48,7 @@ class ExpandOp : public framework::OperatorWithKernel { } // set the first dim to -1 in compile time - if (!ctx->IsRuntime()) { + if (!ctx->IsRuntime() && x_dims[0] < 0) { out_shape[0] = x_dims[0]; }