提交 9e89ae07 编写于 作者: S sunyanfang01

add concat

上级 684c0228
......@@ -467,7 +467,7 @@ class CaffeOpMapper(OpMapper):
def Concat(self, node):
assert len(
node.inputs
) > 1, 'The count of Concat node\'s input is not more than 1.'
) >= 1, 'The count of Concat node\'s input is not more than 1.'
inputs = []
for i in range(len(node.inputs)):
input = self.graph.get_bottom_node(node, idx=i, copy=True)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册