Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix

This commit is contained in:
huangmengxi 2021-07-06 11:17:09 +08:00
parent de832c989a
commit 017809eba1
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mindspore/ops/composite/multitype_ops/_compile_utils.py
View File

@ -706,8 +706,9 @@ def tensor_setitem_by_tuple_with_tensor(data, tuple_index, value):
dim1_start, dim1_stop, _ = const_utils.normalize_slice(tuple_index[1], data.shape[1])
if dim1_stop - dim1_start <= 0:
return data
start = (tuple_index[0], dim1_start)
stop = (tuple_index[0] + 1, dim1_stop)
dim0_start = tuple_index[0] if tuple_index[0] >= 0 else tuple_index[0] + data.shape[0]
start = (dim0_start, dim1_start)
stop = (dim0_start + 1, dim1_stop)
step = (1, 1)
value_shape = (dim1_stop - dim1_start,) + const_utils.tuple_slice(data.shape, 2, None)
value = _broadcast(value_shape, value)