[FRONTEND] Correct the use of `concatenate` operator (#181)
* Correct the use of `concatenate` operator * Optimize
This commit is contained in:
Родитель
c0cde10d57
Коммит
3833ed2ee9
|
@ -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):
|
||||
|
|
Загрузка…
Ссылка в новой задаче