- 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 "RnnStack.hpp"
|
||||||
|
#include "RnnTextHelper.hpp"
|
||||||
|
|
||||||
RnnStack::RnnStack(const std::string &name, size_t numContext)
|
RnnStack::RnnStack(const std::string &name, size_t numContext)
|
||||||
: AStack(StackType::Rnn, name)
|
: 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;
|
int k = getSeqLen()- j - 1;
|
||||||
Layer *pLayer = getLayer(j);
|
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);
|
arma::mat vc = v_to_vc(v, c);
|
||||||
pLayer->gibbs_vh(vc, c);
|
pLayer->gibbs_vh(vc, c);
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user