diff --git a/source/RnnStack.cpp b/source/RnnStack.cpp index 4550a8f..ceb4b33 100644 --- a/source/RnnStack.cpp +++ b/source/RnnStack.cpp @@ -12,6 +12,7 @@ */ #include "RnnStack.hpp" +#include "RnnTextHelper.hpp" RnnStack::RnnStack(const std::string &name, size_t numContext) : AStack(StackType::Rnn, name) @@ -98,7 +99,7 @@ arma::mat RnnStack::step_forward(arma::mat &state, const arma::mat& v_curr) { int k = getSeqLen()- j - 1; Layer *pLayer = getLayer(j); - arma::mat v = arma::join_rows(state.row(k), z); + arma::mat v = arma::join_rows(z, state.row(k)); arma::mat vc = v_to_vc(v, c); pLayer->gibbs_vh(vc, c);