提交 77933bef authored 作者: Adrian Seyboldt's avatar Adrian Seyboldt 提交者: Brandon T. Willard

Add numba sigmoid implementation

上级 0a4a35c7
...@@ -27,6 +27,7 @@ from aesara.scalar.basic import ( ...@@ -27,6 +27,7 @@ from aesara.scalar.basic import (
Second, Second,
Switch, Switch,
) )
from aesara.scalar.math import Sigmoid
@numba_funcify.register(ScalarOp) @numba_funcify.register(ScalarOp)
...@@ -241,3 +242,12 @@ def numba_funcify_Inv(op, node, **kwargs): ...@@ -241,3 +242,12 @@ def numba_funcify_Inv(op, node, **kwargs):
return 1 / x return 1 / x
return inv return inv
@numba_funcify.register(Sigmoid)
def numba_funcify_Sigmoid(op, node, **kwargs):
@numba_basic.numba_njit(inline="always", fastmath=config.numba__fastmath)
def sigmoid(x):
return 1 / (1 + np.exp(-x))
return sigmoid
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论