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

fix the docs of the operator Concat

This commit is contained in:
dinglinhe 2021-06-24 17:16:17 +08:00
parent d621cdbc97
commit 5f1cdf315e
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/ops/_utils/utils.py
View File

@ -84,7 +84,8 @@ def get_concat_offset(x_shp, x_type, axis, prim_name):
validator.check('x_type[%d]' % i, x_type[i], 'x_type[0]', x_type[0], Rel.EQ, prim_name)
for j in range(rank_base):
if j != axis and v[j] != x_shp[0][j]:
raise ValueError(f"For \'{prim_name}\' element {i} shape in input can not concat with first element")
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]}.")
offset.append(all_shp)
if all_shp == -1 or v[axis] == -1:
all_shp = -1