提交 9baf74f6 authored 作者: Samira Shabanian's avatar Samira Shabanian

allclose added

上级 24b54076
...@@ -6140,7 +6140,7 @@ def test_local_expm1(): ...@@ -6140,7 +6140,7 @@ def test_local_expm1():
f_val = f(x_val) f_val = f(x_val)
f_test = function([x], T.expm1(x), mode=theano.compile.get_default_mode().including('local_expm1')) 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)) assert numpy.allclose(f_val, f_test(x_val))
assert any(isinstance(n.op, T.Elemwise) and isinstance(n.op.scalar_op, theano.scalar.basic.Expm1) assert any(isinstance(n.op, T.Elemwise) and isinstance(n.op.scalar_op, theano.scalar.basic.Expm1)
for n in f.maker.fgraph.toposort()) for n in f.maker.fgraph.toposort())
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论