提交 7bb1b48a authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Use floatX for value in test

上级 6da4304d
...@@ -379,7 +379,7 @@ class T_Scan(unittest.TestCase): ...@@ -379,7 +379,7 @@ class T_Scan(unittest.TestCase):
fun = theano.function([inp], [broadcasted_inp, gr]) fun = theano.function([inp], [broadcasted_inp, gr])
# Execute the Theano function and compare outputs to the expected outputs # Execute the Theano function and compare outputs to the expected outputs
inputs = numpy.array([[1, 2], [3, 4]]) inputs = numpy.array([[1, 2], [3, 4]], dtype=theano.config.floatX)
expected_out1 = numpy.repeat(inputs[None], n_steps, axis=0) expected_out1 = numpy.repeat(inputs[None], n_steps, axis=0)
expected_out2 = numpy.ones(inputs.shape, dtype="int8") * n_steps expected_out2 = numpy.ones(inputs.shape, dtype="int8") * n_steps
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论