From c7ee494caff75b9df7a9021f5e7abf5afefebd1b Mon Sep 17 00:00:00 2001 From: zhaoxinxin Date: Thu, 19 Nov 2020 21:50:51 +0800 Subject: [PATCH] modified: ge/host_kernels/strided_slice_kernel.cc --- ge/host_kernels/strided_slice_kernel.cc | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) 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); + } } } }