Browse Source

concat offset update

tags/v0.3.0-alpha
zhaozhenlong 5 years ago
parent
commit
cbb4136b62
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ops/operations/array_ops.py

+ 1
- 1
mindspore/ops/operations/array_ops.py View File

@@ -1316,7 +1316,7 @@ class Concat(PrimitiveWithInfer):
axis = self.axis
x_shp = input_x['shape']
x_type = input_x['dtype']
_, all_shp, _ = _get_concat_offset(x_shp, x_type, axis)
_, all_shp, _ = _get_concat_offset(x_shp, x_type, axis, self.name)
self.add_prim_attr('T', x_type[0].element_type())
self.add_prim_attr('inputNums', len(x_shp))
ret_shp = x_shp[0].copy()


Loading…
Cancel
Save