modified: ge/host_kernels/strided_slice_kernel.cc

pull/339/head
zhaoxinxin 4 years ago
parent c7ee494caf
commit 2a5548b192

@ -306,7 +306,7 @@ void StridedSliceKernel::ExpandStrideWithEllipsisMask(const size_t x_dims_num,
orig_end_vec[i] = x_dims.at(i);
orig_stride_vec[i] = 1;
if (orig_begin_vec.size() < x_dims_num) {
for (auto j = 0; j < (x_dims_num - orig_begin_vec.size() + 1); ++j) {
for (auto j = 1; 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);

Loading…
Cancel
Save