From cad8dc52b2b021081cc2446b693e63cb79231cba Mon Sep 17 00:00:00 2001 From: ZhidanLiu Date: Thu, 8 Dec 2022 16:46:08 +0800 Subject: [PATCH] fix bug in call old version split function in Tensor --- mindspore/python/mindspore/common/tensor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/python/mindspore/common/tensor.py b/mindspore/python/mindspore/common/tensor.py index 43375c616eb..cc44f455fcb 100644 --- a/mindspore/python/mindspore/common/tensor.py +++ b/mindspore/python/mindspore/common/tensor.py @@ -3124,7 +3124,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: