提交 dd75c2d5 authored 作者: Brandon T. Willard's avatar Brandon T. Willard 提交者: Brandon T. Willard

Add missing local_reciprocal_1_plus_exp test

上级 bfabe261
...@@ -19,7 +19,7 @@ from aesara.configdefaults import config ...@@ -19,7 +19,7 @@ from aesara.configdefaults import config
from aesara.graph.basic import Constant from aesara.graph.basic import Constant
from aesara.graph.fg import FunctionGraph from aesara.graph.fg import FunctionGraph
from aesara.graph.opt import LocalOptGroup, TopoOptimizer, check_stack_trace, out2in from aesara.graph.opt import LocalOptGroup, TopoOptimizer, check_stack_trace, out2in
from aesara.graph.opt_utils import is_same_graph from aesara.graph.opt_utils import is_same_graph, optimize_graph
from aesara.graph.optdb import OptimizationQuery from aesara.graph.optdb import OptimizationQuery
from aesara.misc.safe_asarray import _asarray from aesara.misc.safe_asarray import _asarray
from aesara.tensor import inplace from aesara.tensor import inplace
...@@ -4001,6 +4001,13 @@ def test_local_log_sum_exp3(): ...@@ -4001,6 +4001,13 @@ def test_local_log_sum_exp3():
assert np.allclose(optimised_ret, 100.0) assert np.allclose(optimised_ret, 100.0)
def test_local_reciprocal_1_plus_exp():
x = vector("x")
y = aet.reciprocal(1 + exp(x))
z = optimize_graph(y, include=["canonicalization", "stabilize", "specialize"])
assert z.owner.op == sigmoid
class TestSigmoidOpts: class TestSigmoidOpts:
def get_mode(self, excluding=None): def get_mode(self, excluding=None):
""" """
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论