提交 cae31288 authored 作者: gdesjardins's avatar gdesjardins

Testcase for changeset 4479:3258bb36d77a.

Closes ticket #518.
上级 26364b9c
...@@ -2087,6 +2087,13 @@ if __name__ == '__main__': ...@@ -2087,6 +2087,13 @@ if __name__ == '__main__':
# unittest.main() # unittest.main()
test_fusion().tes_memory_leak() test_fusion().tes_memory_leak()
def test_local_mul_to_neg():
"""
Test that a multiplication by -1 or -1.0 yields the appropriate data type
"""
a = T.imatrix()
f1 = theano.function([a], -1*a)
f2 = theano.function([a], -1.0*a)
aval = numpy.random.randint(0,10,(2,2))
assert f1(aval).dtype == a.dtype
assert f2(aval).dtype == 'float64'
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论