From 3833ed2ee97a32fdbf88b562b55f822f5f542e79 Mon Sep 17 00:00:00 2001
From: PENGUINLIONG <admin@penguinliong.moe>
Date: Thu, 19 Oct 2017 02:45:40 +0800
Subject: [PATCH] [FRONTEND] Correct the use of `concatenate` operator (#181)

* Correct the use of `concatenate` operator

* Optimize
---
 nnvm/python/nnvm/frontend/coreml.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/nnvm/python/nnvm/frontend/coreml.py b/nnvm/python/nnvm/frontend/coreml.py
index d8d6461e..e89c784d 100644
--- a/nnvm/python/nnvm/frontend/coreml.py
+++ b/nnvm/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):