Add error message for slice op(#30851)

4 years ago · 97f7a70c01
parent 87197f8c2e
commit 97f7a70c01
1 changed files with 7 additions and 0 deletions
--- a/paddle/fluid/operators/slice_op.cc
+++ b/paddle/fluid/operators/slice_op.cc
@ -121,6 +121,13 @@ class SliceOp : public framework::OperatorWithKernel {
          start = std::max(start, 0);
          end = std::max(end, 0);
          end = std::min(end, dim_value);
+
+          PADDLE_ENFORCE_LE(start, dim_value,
+                            platform::errors::InvalidArgument(
+                                "start should be less than or equal to the "
+                                "dimension value, but received "
+                                "start = %d, shape[%d] = %d.",
+                                starts[i], axes[i], out_dims[axes[i]]));
          PADDLE_ENFORCE_GT(end, start,
                            platform::errors::InvalidArgument(
                                "end should greater than start, but received "