提交 a92be852 authored 作者: Samira Shabanian's avatar Samira Shabanian

compile mode fixed

上级 f244431b
......@@ -6138,7 +6138,7 @@ def test_local_expm1():
r = function([u], s)
x_val = numpy.random.rand(4, 3).astype(config.floatX)
f_val = f(x_val)
f_test = function([x], T.expm1(x))
f_test = function([x], T.expm1(x), mode=theano.compile.get_default_mode().including('local_expm1'))
assert numpy.all(f_val == f_test(x_val))
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论