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

Added astype(theano.config.floatX)

上级 28fd9366
......@@ -109,7 +109,7 @@ class TestKeepDims(unittest.TestCase):
def test_norm(self):
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")
for axis in [0, 1, 2, [0], [1], [2], None,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论