提交 b58395fc authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Fix output_storage in compute_test_value, when there are more than 1 inputs

上级 0011ec8a
......@@ -353,7 +353,7 @@ class PureOp(object):
if run_perform:
# compute output value once with test inputs to validate graph
output_storage = [[None] * len(node.outputs)]
output_storage = [[None]] * len(node.outputs)
node.op.perform(node, input_vals, output_storage)
# add 'test_value' to output tags, so that downstream ops can use these
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论