diff options
Diffstat (limited to 'ethosu/vela/numeric_util.py')
-rw-r--r-- | ethosu/vela/numeric_util.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/ethosu/vela/numeric_util.py b/ethosu/vela/numeric_util.py index 4ebef8e5..3d26444a 100644 --- a/ethosu/vela/numeric_util.py +++ b/ethosu/vela/numeric_util.py @@ -77,13 +77,17 @@ def clamp_tanh(x): return y +def sigmoid(x): + return 1 / (1 + math.exp(-x)) + + def clamp_sigmoid(x): if x <= -8: y = 0.0 elif x >= 8: y = 1.0 else: - y = 1 / (1 + math.exp(-x)) + y = sigmoid(x) return y |