提交 4d1293e4 authored 作者: Benjamin Scellier's avatar Benjamin Scellier 提交者: Nicolas Ballas

file theano/gof/opt.py

上级 b5f5529b
...@@ -15,7 +15,7 @@ import time ...@@ -15,7 +15,7 @@ import time
import warnings import warnings
import traceback import traceback
import numpy import numpy as np
import theano import theano
from theano import config from theano import config
...@@ -1695,8 +1695,7 @@ class PatternSub(LocalOptimizer): ...@@ -1695,8 +1695,7 @@ class PatternSub(LocalOptimizer):
u = u.merge(expr, v) u = u.merge(expr, v)
elif (isinstance(pattern, (integer_types, float)) and elif (isinstance(pattern, (integer_types, float)) and
isinstance(expr, graph.Constant)): isinstance(expr, graph.Constant)):
if numpy.all( if np.all(theano.tensor.constant(pattern).value == expr.value):
theano.tensor.constant(pattern).value == expr.value):
return u return u
else: else:
return retry_with_equiv() return retry_with_equiv()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论