fix wrong error info raised

This commit is contained in:
liuxiao93 2022-01-25 09:07:23 +08:00
parent f93ef37aad
commit 9a35d47d30
1 changed files with 1 additions and 1 deletions

View File

@ -100,7 +100,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] = {x_shp[0]} and shape[1] = {x_shp[1]}.")
f"shape[0] = {x_shp[0]} and shape[{i}] = {x_shp[i]}.")
offset.append(all_shp)
if all_shp == -1 or v[axis] == -1:
all_shp = -1