From f2a1b79e993c3d9c8ec529f06e304532487b13bb Mon Sep 17 00:00:00 2001 From: Ryan Curtin Date: Sun, 27 Mar 2022 14:16:58 -0400 Subject: [PATCH] Fix incorrect step reference. --- src/mlpack/methods/ann/layer/lstm_impl.hpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/mlpack/methods/ann/layer/lstm_impl.hpp b/src/mlpack/methods/ann/layer/lstm_impl.hpp index b420c4d522..b4cb5a6048 100644 --- a/src/mlpack/methods/ann/layer/lstm_impl.hpp +++ b/src/mlpack/methods/ann/layer/lstm_impl.hpp @@ -267,7 +267,7 @@ void LSTMType::Forward( if (this->HasPreviousStep()) { outputGate += - output2GateOutputWeight * outParameter.slice(this->CurrentStep()); + output2GateOutputWeight * outParameter.slice(this->PreviousStep()); } outputGate.each_col() += input2GateOutputBias;