Update src/mlpack/methods/ann/rnn_impl.hpp
Co-authored-by: Ryan Curtin <ryan@ratml.org>
This commit is contained in:
co-authored by
Ryan Curtin
parent
3164cf68b2
commit
f229e4b2d1
@@ -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);
|
||||
|
||||
|
||||
Reference in New Issue
Block a user