提交 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
from aesara.graph.basic import Constant
from aesara.graph.fg import FunctionGraph
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.misc.safe_asarray import _asarray
from aesara.tensor import inplace
......@@ -4001,6 +4001,13 @@ def test_local_log_sum_exp3():
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:
def get_mode(self, excluding=None):
"""
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论