Lichen455
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Concat operator prompts incorrectly

This commit is contained in:
dinglinhe 2021-06-29 11:09:40 +08:00
parent 4b2f838fb4
commit 5d30875b27
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/ops/_utils/utils.py
View File

@ -85,7 +85,7 @@ def get_concat_offset(x_shp, x_type, axis, prim_name):
for j in range(rank_base):
if j != axis and v[j] != x_shp[0][j]:
raise ValueError(f"The shape of the two input elements of the Concat operator do not match:"
f"shape[0] = {v} and shape[1] = {x_shp[0]}.")
f"shape[0] = {x_shp[0]} and shape[1] = {x_shp[1]}.")
offset.append(all_shp)
if all_shp == -1 or v[axis] == -1:
all_shp = -1