提交 25ba75cb authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix the 0-dim test to use the proper op.

上级 da2d163e
...@@ -161,7 +161,7 @@ class T_subtensor(unittest.TestCase, utt.TestOptimizationMixin): ...@@ -161,7 +161,7 @@ class T_subtensor(unittest.TestCase, utt.TestOptimizationMixin):
def test1_0_dims(self): def test1_0_dims(self):
n = self.shared(numpy.ones((), dtype=self.dtype)) n = self.shared(numpy.ones((), dtype=self.dtype))
t = theano.tensor.Subtensor([])(n) t = self.sub([])(n)
self.assertTrue(isinstance(t.owner.op, Subtensor)) self.assertTrue(isinstance(t.owner.op, Subtensor))
mode = self.mode mode = self.mode
self.mode = mode.excluding("local_useless_subtensor") self.mode = mode.excluding("local_useless_subtensor")
...@@ -188,7 +188,6 @@ class T_subtensor(unittest.TestCase, utt.TestOptimizationMixin): ...@@ -188,7 +188,6 @@ class T_subtensor(unittest.TestCase, utt.TestOptimizationMixin):
self.assertTrue(tval == 5.0) self.assertTrue(tval == 5.0)
def test1_ok_range_infinite(self): def test1_ok_range_infinite(self):
#Subtensor.debug = True
n = self.shared(numpy.arange(3, dtype=self.dtype)) n = self.shared(numpy.arange(3, dtype=self.dtype))
t = n[1:] t = n[1:]
self.assertTrue(isinstance(t.owner.op, Subtensor)) self.assertTrue(isinstance(t.owner.op, Subtensor))
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论