Fixing example

jatinchowdhury18 · Nov 14, 2023 · 4a3e424 · 4a3e424
1 parent 1ef0dae
commit 4a3e424
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/examples/custom_layer_model/GatedActivation_STL.h b/examples/custom_layer_model/GatedActivation_STL.h
@@ -27,8 +27,13 @@ class GatedActivation
     /** Performs forward propagation for gated activation. */
     inline void forward(const T (&ins)[in_size]) noexcept
     {
+        const auto sigmoid = [] (T x)
+        {
+            return (T) 1 / ((T) 1 + std::exp (-x));
+        };
+
         for(int i = 0; i < out_size; ++i)
-            outs[i] = std::tanh(ins[i]) * RTNeural::sigmoid(ins[i + out_size]);
+            outs[i] = std::tanh(ins[i]) * sigmoid(ins[i + out_size]);
     }
 
     T outs alignas(RTNEURAL_DEFAULT_ALIGNMENT)[out_size];