提交 9b8e93e7 authored 作者: Frederic's avatar Frederic

Don't raise an error for test value when the compute_test_value is off during opt.

上级 31c4f10b
...@@ -4656,9 +4656,10 @@ def local_elemwise_fusion_op(OP, max_input_fct=lambda node: 1024): ...@@ -4656,9 +4656,10 @@ def local_elemwise_fusion_op(OP, max_input_fct=lambda node: 1024):
else: else:
s = scalar.Scalar(i.dtype).make_variable() s = scalar.Scalar(i.dtype).make_variable()
try: try:
v = gof.op.get_test_value(i) if theano.config.compute_test_value != 'off':
if v.size > 0: v = gof.op.get_test_value(i)
s.tag.test_value = gof.op.get_test_value(i).flatten()[0] if v.size > 0:
s.tag.test_value = v.flatten()[0]
except AttributeError: except AttributeError:
pass pass
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论