- forward: provide current expect next

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@836 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
2022-01-18 20:14:31 +00:00
parent c09e92984a
commit 1c206de402
+2 -1
View File
@@ -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);