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

!14738 Change Tensor zero dimension check code to make it faster. 

From: @liangzhibo
Reviewed-by: @ginfung,@zh_qh
Signed-off-by: @zh_qh
This commit is contained in:
mindspore-ci-bot 2021-04-07 20:49:41 +08:00 committed by Gitee
parent 3b87d26655 2ce5de2f47
commit 13d9ad0f2a
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
mindspore/common/tensor.py
View File

@ -92,9 +92,13 @@ class Tensor(Tensor_):
if isinstance(shape, numbers.Number):
shape = (shape,)
if input_data is not None and isinstance(input_data, (tuple, list, np.ndarray)) \
and np.array(input_data).ndim > 1 and np.array(input_data).size == 0:
raise ValueError("input_data can not contain zero dimension.")
if input_data is not None:
if isinstance(input_data, np.ndarray) and input_data.ndim > 1 and input_data.size == 0:
raise ValueError("input_data can not contain zero dimension.")
if isinstance(input_data, (tuple, list)) and np.array(input_data).ndim > 1 \
and np.array(input_data).size == 0:
raise ValueError("input_data can not contain zero dimension.")
if shape is not None and not (hasattr(init, "__enable_zero_dim__") and init.__enable_zero_dim__):
if 0 in shape:
raise ValueError("Shape can not contain zero value.")