fix elu activation function (tflearn#853)

ConnieTong · Jul 29, 2017 · d28030a · d28030a
1 parent 6b05603
commit d28030a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tflearn/activations.py b/tflearn/activations.py
@@ -303,4 +303,4 @@ def selu(x):
     """
     alpha = 1.6732632423543772848170429916717
     scale = 1.0507009873554804934193349852946
-    return scale * tf.nn.elu(x, alpha)
+    return scale*tf.where(x>=0.0, x, alpha*tf.nn.elu(x))