提交 e53c49ab authored 作者: Ricardo's avatar Ricardo 提交者: Ricardo Vieira

Add test for constraint in `log1msigm_to_softplus` rewrite

上级 983df069
...@@ -4376,6 +4376,14 @@ class TestSoftplusOpts: ...@@ -4376,6 +4376,14 @@ class TestSoftplusOpts:
# assert check_stack_trace(f, ops_to_check='all') # assert check_stack_trace(f, ops_to_check='all')
f(np.random.random((54, 11)).astype(config.floatX)) f(np.random.random((54, 11)).astype(config.floatX))
# Test close to 1
out = log(1.000001 - sigmoid(x))
f = aesara.function([x], out, mode=self.m)
topo = f.maker.fgraph.toposort()
assert len(topo) == 2
assert isinstance(topo[0].op.scalar_op, aesara.scalar.Softplus)
assert isinstance(topo[1].op.scalar_op, aesara.scalar.Neg)
# Same test with a flatten # Same test with a flatten
out = log(1 - aet.flatten(sigmoid(x))) out = log(1 - aet.flatten(sigmoid(x)))
f = aesara.function([x], out, mode=self.m) f = aesara.function([x], out, mode=self.m)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论