From 8f89db7fb6be7c0a67768003f53243fab5f1f5bf Mon Sep 17 00:00:00 2001 From: Adam Kropp <30504149+akropp@users.noreply.github.com> Date: Mon, 13 Nov 2023 14:28:12 -0500 Subject: [PATCH] Update src/mlpack/methods/ann/activation_functions/lisht_function.hpp Co-authored-by: Ryan Curtin --- .../methods/ann/activation_functions/lisht_function.hpp | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/mlpack/methods/ann/activation_functions/lisht_function.hpp b/src/mlpack/methods/ann/activation_functions/lisht_function.hpp index d0a38f9a0c..99f21553c6 100644 --- a/src/mlpack/methods/ann/activation_functions/lisht_function.hpp +++ b/src/mlpack/methods/ann/activation_functions/lisht_function.hpp @@ -84,7 +84,9 @@ class LiSHTFunction * @param dy The resulting derivatives. */ template - static void Deriv(const InputVecType &x, const OutputVecType &/* y */, DerivVecType &dy) + static void Deriv(const InputVecType& x, + const OutputVecType& /* y */, + DerivVecType& dy) { dy = arma::tanh(x) + x % (1 - arma::pow(arma::tanh(x), 2)); }