fix bug of infer shape for slice (PaddlePaddle#43443)

zyfncg · Jun 14, 2022 · dcdcbb6 · dcdcbb6
1 parent efb6fa7
commit dcdcbb6
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/paddle/fluid/operators/slice_op.cc b/paddle/fluid/operators/slice_op.cc
@@ -101,7 +101,11 @@ class SliceOp : public framework::OperatorWithKernel {
           platform::errors::InvalidArgument(
               "The size of ends must be equal to the size of axes."));
     }
-
+    for (auto &axis : axes) {
+      if (axis < 0) {
+        axis = std::max(0, axis + in_dims.size());
+      }
+    }
     phi::funcs::CheckAndUpdateSliceAttrs<int>(in_dims, axes, &starts, &ends,
                                               nullptr, &infer_flags);