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

!46592 fix bug in call old version split function in Tensor 

Merge pull request !46592 from ZhidanLiu/fixbug
This commit is contained in:
i-robot 2022-12-09 06:57:59 +00:00 committed by Gitee
parent 4a2ffb1b9d cad8dc52b2
commit d7a19bf0fe
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/python/mindspore/common/tensor.py
View File

@ -3104,7 +3104,7 @@ class Tensor(Tensor_):
raise ValueError(f"For 'Tensor.repeat', the length of 'repeats' must be the same as the shape of the "
f"original tensor in the 'axis' dimension, but got the length of 'repeats' "
f"{len(repeats)}, the shape of the original tensor in the 'axis' dimension {size}.")
subs = tensor_operator_registry.get('split')(axis, size)(input_x)
subs = tensor_operator_registry.get('tensor_split')(input_x, size, axis)
repeated_subs = []
for sub, rep in zip(subs, repeats):
if rep != 0: