提交 5f58b6fc authored 作者: Frederic's avatar Frederic

fix test when floatX=float32.

上级 b78da17b
...@@ -507,7 +507,7 @@ class test_Eigh(test_Eig): ...@@ -507,7 +507,7 @@ class test_Eigh(test_Eig):
op = staticmethod(eigh) op = staticmethod(eigh)
def test_uplo(self): def test_uplo(self):
S = self.S S = self.S
a = theano.tensor.matrix() a = theano.tensor.matrix(dtype=self.dtype)
wu, vu = [out.eval({a: S}) for out in self.op(a, 'U')] wu, vu = [out.eval({a: S}) for out in self.op(a, 'U')]
wl, vl = [out.eval({a: S}) for out in self.op(a, 'L')] wl, vl = [out.eval({a: S}) for out in self.op(a, 'L')]
assert_array_almost_equal(wu, wl) assert_array_almost_equal(wu, wl)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论