Fix expand op compile time bug

test=develop

Fix expand op compile time bug
test=develop
bc3e0d6e · minqiyang · e2130502 · bc3e0d6e
显示空白变更内容
内联并排

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
@@ -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];
    }