提交 84c6fa49 authored 作者: Samira Shabanian's avatar Samira Shabanian

Added astype(theano.config.floatX)

上级 28fd9366
...@@ -109,7 +109,7 @@ class TestKeepDims(unittest.TestCase): ...@@ -109,7 +109,7 @@ class TestKeepDims(unittest.TestCase):
def test_norm(self): def test_norm(self):
x = tensor.dtensor3() x = tensor.dtensor3()
a = numpy.random.rand(3, 2, 4) a = numpy.random.rand(3, 2, 4).astype(theano.config.floatX)
mode = theano.compile.Mode(optimizer="fast_compile", linker="py") mode = theano.compile.Mode(optimizer="fast_compile", linker="py")
for axis in [0, 1, 2, [0], [1], [2], None, for axis in [0, 1, 2, [0], [1], [2], None,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论