提交 60f4eb55 authored 作者: Frederic Bastien's avatar Frederic Bastien

fix test in FAST_COMPILE mode.

上级 f6f4faeb
......@@ -1531,23 +1531,23 @@ def test_constant_get_stabilized():
#When this error is fixed, the following line should be ok.
assert f()==800,f()
class T_local_sum_canonicalize(unittest.TestCase):
class T_local_sum(unittest.TestCase):
def setUp(self):
self.mode = theano.compile.get_default_mode().including('canonicalize')
def test_local_sum_all_to_none(self):
a = T.tensor3()
input=numpy.arange(3*3*3).reshape(3,3,3)
f = theano.function([a],a.sum())
f = theano.function([a],a.sum()),mode=self.mode)
assert len(f.maker.env.nodes)==1
assert numpy.allclose(f(input),input.sum())
f = theano.function([a],a.sum([0,1,2]))
f = theano.function([a],a.sum([0,1,2]),mode=self.mode)
assert len(f.maker.env.nodes)==1
assert numpy.allclose(f(input),input.sum())
f = theano.function([a],a.sum(0).sum(0).sum(0))
f = theano.function([a],a.sum(0).sum(0).sum(0),mode=self.mode)
assert len(f.maker.env.nodes)==1
assert numpy.allclose(f(input),input.sum())
......@@ -1557,19 +1557,19 @@ class T_local_sum_canonicalize(unittest.TestCase):
dims=[(0,0),(1,0),(2,0),(0,1),(1,1),(2,1)]
for d,dd in dims:
f = theano.function([a],a.sum(d).sum(dd))
f = theano.function([a],a.sum(d).sum(dd),mode=self.mode)
assert numpy.allclose(f(input),input.sum(d).sum(dd))
assert len(f.maker.env.nodes)==1
for d,dd in dims:
f = theano.function([a],a.sum(d).sum(dd).sum(0))
f = theano.function([a],a.sum(d).sum(dd).sum(0),mode=self.mode)
assert numpy.allclose(f(input),input.sum(d).sum(dd).sum(0))
assert len(f.maker.env.nodes)==1
for d in [0,1,2]:
f = theano.function([a],a.sum(d).sum(None))
f = theano.function([a],a.sum(d).sum(None),mode=self.mode)
assert numpy.allclose(f(input),input.sum(d).sum())
assert len(f.maker.env.nodes)==1
for d in [0,1,2]:
f = theano.function([a],a.sum(None).sum())
f = theano.function([a],a.sum(None).sum(),mode=self.mode)
assert numpy.allclose(f(input),input.sum())
assert len(f.maker.env.nodes)==1
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论