提交 116eecaa authored 作者: --global's avatar --global

make test compatible with floatX=float32

上级 80e2bba5
...@@ -3058,9 +3058,9 @@ class T_Scan(unittest.TestCase): ...@@ -3058,9 +3058,9 @@ class T_Scan(unittest.TestCase):
f_ref = theano.function([W1, W2, step_indices], o, mode='FAST_COMPILE') f_ref = theano.function([W1, W2, step_indices], o, mode='FAST_COMPILE')
# Compare the results of the two implementations # Compare the results of the two implementations
input_values = [numpy.random.random((5, 5)), input_values = [numpy.random.random((5, 5)).astype("float32"),
numpy.random.random((5, 5)), numpy.random.random((5, 5)).astype("float32"),
numpy.arange(5)] numpy.arange(5).astype("float32")]
out = f(*input_values) out = f(*input_values)
out_ref = f_ref(*input_values) out_ref = f_ref(*input_values)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论