Change backward op infershape

test=develop

Change backward op infershape
test=develop
c4cf5967 · minqiyang · bc3e0d6e · c4cf5967
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddle/fluid/operators/expand_op.cc paddle/fluid/operators/expand_op.cc +1 -1

未找到文件。
--- a/paddle/fluid/operators/expand_op.cc
+++ b/paddle/fluid/operators/expand_op.cc
@@ -115,7 +115,7 @@ class ExpandGradOp : public framework::OperatorWithKernel {
    auto out_dims = ctx->GetInputDim(framework::GradVarName("Out"));

    size_t start_pos = 0u;
-    if (!ctx->IsRuntime()) {
+    if (!ctx->IsRuntime() && x_dims[0] < 0) {
      PADDLE_ENFORCE_EQ(
          x_dims[0], out_dims[0],
          "The first dimension size of Input(Out@GRAD) should be "