提交 93f46dfd authored 作者: Ian Goodfellow's avatar Ian Goodfellow

added test for bug fixed in commit 4478:c77bf5ba5822

上级 4b94629c
......@@ -790,6 +790,16 @@ def test_many_arg_elemwise():
assert numpy.allclose(results_gpu, results_cpu)
def test_duplicate_arg_elemwise():
A = theano.tensor.fmatrix()
B = A + A
f = theano.function([A],B, mode = mode_with_gpu)
Aval = numpy.random.RandomState([1,2,3]).randn(5,5)
Bval = Aval + Aval
assert numpy.allclose(Bval,f(Aval))
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论