提交 b5e7e1d9 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Fix test

上级 3be0a29f
...@@ -993,7 +993,7 @@ class Test_rebuild_strict(unittest.TestCase): ...@@ -993,7 +993,7 @@ class Test_rebuild_strict(unittest.TestCase):
x, y = tensor.ivectors('x', 'y') x, y = tensor.ivectors('x', 'y')
z = x * y z = x * y
f = theano.function([w, y], z, givens=[(x, w)], rebuild_strict=False) f = theano.function([w, y], z, givens=[(x, w)], rebuild_strict=False)
z_val = f(numpy.ones((3, 5), dtype='int32'), numpy.arange(5)) z_val = f(numpy.ones((3, 5), dtype='int32'), numpy.arange(5, dtype='int32'))
assert z_val.ndim == 2 assert z_val.ndim == 2
assert numpy.all(z_val == numpy.ones((3, 5)) * numpy.arange(5)) assert numpy.all(z_val == numpy.ones((3, 5)) * numpy.arange(5))
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论