提交 6291e64b authored 作者: Olivier Breuleux's avatar Olivier Breuleux

fixed Op.__call__ to use the default_output field

上级 3650c6e8
...@@ -26,6 +26,9 @@ class Op(object2): ...@@ -26,6 +26,9 @@ class Op(object2):
def __call__(self, *inputs): def __call__(self, *inputs):
node = self.make_node(*inputs) node = self.make_node(*inputs)
if self.default_output is not None:
return node.outputs[self.default_output]
else:
if len(node.outputs) == 1: if len(node.outputs) == 1:
return node.outputs[0] return node.outputs[0]
else: else:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论