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

file theano/gof/opt.py

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