xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!18541 add non negative validation of begin args for slice operator. 

Merge pull request !18541 from wangshuide/wsd_master
This commit is contained in:
i-robot 2021-06-21 14:45:15 +08:00 committed by Gitee
parent 205e368b73 c4014a8e2d
commit 9125f647f3
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/ops/operations/array_ops.py
View File

@ -2653,7 +2653,7 @@ class Slice(PrimitiveWithInfer):
Inputs:
- **input_x** (Tensor): The target tensor.
- **begin** (Union[tuple, list]): The beginning of the slice. Only constant value is allowed.
- **begin** (Union[tuple, list]): The beginning of the slice. Only constant value(>=0) is allowed.
- **size** (Union[tuple, list]): The size of the slice. Only constant value is allowed.
Outputs:
@ -2709,6 +2709,7 @@ class Slice(PrimitiveWithInfer):
'len x\'s dim', x_shp_len)
for i in range(x_shp_len):
validator.check_positive_int(size_v[i], f'input size[{i}]')
validator.check_non_negative_int(begin_v[i], f'input begin[{i}]')
if x_shape[i] < begin_v[i] + size_v[i]:
y = begin_v[i] + size_v[i]
raise ValueError("For '%s' slice shape can not bigger than origin shape %d, %d." %