Skip to content

Commit 4fa67f8

Browse files
committed
[strided_slice] handle negative strided_begin before shrink_axis_mask_
1 parent 553c625 commit 4fa67f8

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

src/caffe/layers/strided_slice_layer.cpp

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,12 @@ void StridedSliceLayer<Dtype>::LayerSetUp(const vector<Blob<Dtype> *> &bottom,
3333
strides_.assign(strided_begin_.size(), 1);
3434
}
3535

36+
for (int i = 0; i < strided_begin_.size(); ++i) {
37+
if (strided_begin_[i] < 0) {
38+
strided_begin_[i] = std::max(strided_begin_[i] + bottom[0]->shape(i), 0);
39+
}
40+
}
41+
3642
// add onnx operator Slice parameter: axes
3743
axes_.clear();
3844
std::copy(strided_slice_param.axes().begin(),

0 commit comments

Comments
 (0)