Transpose in activation_layer

Co-authored-by: Amrita Goswami <amrita16thaug646@gmail.com>
seldon-code · Oct 19, 2023 · 56851bd · 56851bd
1 parent 2799742
commit 56851bd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/include/activation_layer.hpp b/include/activation_layer.hpp
@@ -20,7 +20,7 @@ class ActivationLayer : public Layer<scalar>
     // computes dE/dX for a given dE/dY (and update parameters if any)
     Vector<scalar> backward_propagation( Vector<scalar> & output_error, scalar learning_rate ) override
     {
-        return Activation::df( this->input ) * output_error;
+        return Activation::df( this->input ).transpose() * output_error;
     }
 };
 } // namespace Robbie