Update src/mlpack/methods/ann/rnn_impl.hpp

Co-authored-by: Ryan Curtin <ryan@ratml.org>
This commit is contained in:
Omar Shrit
2024-04-26 17:22:25 +02:00
committed by GitHub
co-authored by Ryan Curtin
parent 3164cf68b2
commit f229e4b2d1
+2 -1
View File
@@ -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);