提交 922982a9 authored 作者: Frederic Bastien's avatar Frederic Bastien

make test more robust for futur change.

上级 639178fd
...@@ -27,6 +27,8 @@ class T_sigmoid_opts(unittest.TestCase): ...@@ -27,6 +27,8 @@ class T_sigmoid_opts(unittest.TestCase):
m = theano.config.mode m = theano.config.mode
if m == 'FAST_COMPILE': if m == 'FAST_COMPILE':
m = 'FAST_RUN' m = 'FAST_RUN'
m = theano.compile.mode.get_mode(m)
m = m.excluding('local_elemwise_fusion')
x = T.dvector() x = T.dvector()
...@@ -42,12 +44,8 @@ class T_sigmoid_opts(unittest.TestCase): ...@@ -42,12 +44,8 @@ class T_sigmoid_opts(unittest.TestCase):
# tests inv_1_plus_exp with neg # tests inv_1_plus_exp with neg
f = theano.function([x], T.fill(x,-1.0) / (1+T.exp(-x)), mode=m) f = theano.function([x], T.fill(x,-1.0) / (1+T.exp(-x)), mode=m)
#theano.printing.debugprint(f) assert [node.op for node in f.maker.env.toposort()] == [sigmoid,
assert len(f.maker.env.toposort())==1 T.inplace.neg_inplace]
assert str(f.maker.env.toposort()[0].op)=='Elemwise{Composite{scalar_sigmoid,neg}}'
#without fusion
#assert [node.op for node in f.maker.env.toposort()] == [sigmoid,
# T.inplace.neg_inplace]
# tests double inv_1_plus_exp with neg # tests double inv_1_plus_exp with neg
# (-1)(exp(x)) / (1+exp(x))(1+exp(-x)) # (-1)(exp(x)) / (1+exp(x))(1+exp(-x))
...@@ -55,12 +53,8 @@ class T_sigmoid_opts(unittest.TestCase): ...@@ -55,12 +53,8 @@ class T_sigmoid_opts(unittest.TestCase):
# = - (sigm(x) * sigm(x)) # = - (sigm(x) * sigm(x))
f = theano.function([x], (T.fill(x,-1.0)*T.exp(x)) / ((1+T.exp(x))*(1+T.exp(-x))), mode=m) f = theano.function([x], (T.fill(x,-1.0)*T.exp(x)) / ((1+T.exp(x))*(1+T.exp(-x))), mode=m)
theano.printing.debugprint(f) theano.printing.debugprint(f)
assert len(f.maker.env.toposort())==2 assert [node.op for node in f.maker.env.toposort()] == [sigmoid,
assert f.maker.env.toposort()[0].op == sigmoid T.mul, T.inplace.neg_inplace]
assert str(f.maker.env.toposort()[1].op)=='Elemwise{Composite{mul,neg}}'
#without fusion
#assert [node.op for node in f.maker.env.toposort()] == [sigmoid,
# T.mul, T.inplace.neg_inplace]
def test_1msigmoid(self): def test_1msigmoid(self):
if not register_local_1msigmoid: if not register_local_1msigmoid:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论