提交 fbfee06e authored 作者: David Warde-Farley's avatar David Warde-Farley

Assert right number of arguments in make_node.

上级 a6e40db0
......@@ -155,6 +155,10 @@ class IfElse(PureOp):
return out_shapes
def make_node(self, c, *args):
assert len(args) == 2 * self.n_outs, (
"Wrong number of arguments to make_node: "
"expected %d, got %d" % (2 * self.n_outs, len(args))
)
if not self.gpu:
# When gpu is true, we are given only cuda ndarrays, and we want
# to keep them be cuda ndarrays
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论