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

* Correct the use of `concatenate` operator * Optimize
eric-haibin-lin · 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):