Skip to content
Snippets Groups Projects
Commit 3833ed2e authored by PENGUINLIONG's avatar PENGUINLIONG Committed by Tianqi Chen
Browse files

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

* Correct the use of `concatenate` operator

* Optimize
parent c0cde10d
No related branches found
No related tags found
No related merge requests found
......@@ -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):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment