提交 5965057f authored 作者: abalkin's avatar abalkin 提交者: Frederic

Fixed type of test matrix in test_Eigh.

上级 0143ddf1
......@@ -477,7 +477,7 @@ class test_Eig(utt.InferShapeTester):
def setUp(self):
super(test_Eig, self).setUp()
self.rng = numpy.random.RandomState(utt.fetch_seed())
self.A = theano.tensor.matrix()
self.A = theano.tensor.matrix(dtype=self.dtype)
X = numpy.asarray(self.rng.rand(5, 5),
dtype=self.dtype)
self.S = X.dot(X.T)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论