diff --git a/ge/host_kernels/strided_slice_kernel.cc b/ge/host_kernels/strided_slice_kernel.cc index d59202d2..c73d7c7f 100644 --- a/ge/host_kernels/strided_slice_kernel.cc +++ b/ge/host_kernels/strided_slice_kernel.cc @@ -305,10 +305,12 @@ void StridedSliceKernel::ExpandStrideWithEllipsisMask(const size_t x_dims_num, orig_begin_vec[i] = 0; orig_end_vec[i] = x_dims.at(i); orig_stride_vec[i] = 1; - if (auto j = 0; j < (x_dims_num - orig_begin_vec.size() + 1); ++j) { - orig_begin_vec.insert((orig_begin_vec.begin() + ellipsis_dim + j), 0); - orig_end_vec.insert((orig_end_vec.begin() + ellipsis_dim + j), x_dims.at(ellipsis_dim +j)); - orig_stride_vec.insert((orig_begin_vec.begin() + ellipsis_dim + j), 1); + if (orig_begin_vec.size() < x_dims_num) { + for (auto j = 0; j < (x_dims_num - orig_begin_vec.size() + 1); ++j) { + orig_begin_vec.insert((orig_begin_vec.begin() + ellipsis_dim + j), 0); + orig_end_vec.insert((orig_end_vec.begin() + ellipsis_dim + j), x_dims.at(ellipsis_dim +j)); + orig_stride_vec.insert((orig_begin_vec.begin() + ellipsis_dim + j), 1); + } } } }