sigmoid alias

FluxML · Nov 23, 2017 · 38ba025 · 38ba025
1 parent 18ef20a
commit 38ba025
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/src/NNlib.jl b/src/NNlib.jl
@@ -2,7 +2,7 @@ module NNlib
 
 using Requires
 
-export σ, relu, leakyrelu, elu, swish, selu, softplus, softsign,
+export σ, sigmoid, relu, leakyrelu, elu, swish, selu, softplus, softsign,
   softmax, conv2d, pool
 
 const libnnlib = Libdl.find_library(is_windows() ? "conv.dll" : "conv.so", [@__DIR__])

diff --git a/src/activation.jl b/src/activation.jl
@@ -24,6 +24,8 @@ function.
 """
 σ(x) = one(x) / (one(x) + exp(-x))
 
+const sigmoid = σ
+
 # ForwardDiff numerical stability hack
 σ_stable(x) = ifelse(x < -80, zero(x), one(x) / (one(x) + exp(-x)))