From f229e4b2d13d6eca949403a2e196e38e8f86d876 Mon Sep 17 00:00:00 2001 From: Omar Shrit Date: Fri, 26 Apr 2024 17:22:25 +0200 Subject: [PATCH] Update src/mlpack/methods/ann/rnn_impl.hpp Co-authored-by: Ryan Curtin --- src/mlpack/methods/ann/rnn_impl.hpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/mlpack/methods/ann/rnn_impl.hpp b/src/mlpack/methods/ann/rnn_impl.hpp index 0eede8ac68..1f1b7dd525 100644 --- a/src/mlpack/methods/ann/rnn_impl.hpp +++ b/src/mlpack/methods/ann/rnn_impl.hpp @@ -447,7 +447,8 @@ typename MatType::elem_type RNN< const size_t responseStep = (single) ? 0 : t; MakeAlias(responseData, responses.slice(responseStep), - responses.n_rows, batchSize, begin * responses.slice(responseStep).n_rows); + responses.n_rows, batchSize, + begin * responses.slice(responseStep).n_rows); loss += network.outputLayer.Forward(outputData, responseData);