- 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:
+2
-1
@@ -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);
|
||||
|
||||
|
||||
Reference in New Issue
Block a user