Fix expand op compile time bug

test=develop
6 years ago · bc3e0d6e01
parent e213050223
commit bc3e0d6e01
1 changed files with 1 additions and 1 deletions
--- 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];
    }