[FRONTEND] Correct the use of concatenate operator (#181)

* Correct the use of `concatenate` operator * Optimize
dmlc · Oct 18, 2017 · 8d79cfd · 8d79cfd
1 parent a96133e
commit 8d79cfd
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/python/nnvm/frontend/coreml.py b/python/nnvm/frontend/coreml.py
@@ -246,9 +246,7 @@ def ConcatLayerParams(op, insyms, symtab):
  insyms = [insyms]
  if op.sequenceConcat:
  raise NotImplementedError("Sequence Concat not supported")
- ret = insyms[0]
- for i in range(1, len(insyms)):
- ret = _sym.concat(ret, insyms[i], dim=1)
+ ret = _sym.concatenate(*insyms, dim=1)
  return ret
 
 def FlattenLayerParams(op, insym, symtab):