提交 dfd24e90 authored 作者: Ilya Kulikov's avatar Ilya Kulikov

cast ndarray to theano floatX type

上级 514908e1
...@@ -525,7 +525,7 @@ def test_tensorinv(): ...@@ -525,7 +525,7 @@ def test_tensorinv():
X = tensorinv(A) X = tensorinv(A)
tf = function([A], [X]) tf = function([A], [X])
a = numpy.eye(4 * 6) a = numpy.eye(4 * 6).astype(theano.config.floatX)
a.shape = (4, 6, 8, 3) a.shape = (4, 6, 8, 3)
n_ainv = numpy.linalg.tensorinv(a) n_ainv = numpy.linalg.tensorinv(a)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论