- improved RnnStack

- AStack: context doesn't belong o training data

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@827 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
2022-01-18 14:27:39 +00:00
parent 17770f402c
commit 9a8ad70a0b
4 changed files with 99 additions and 38 deletions
+14 -1
View File
@@ -18,6 +18,8 @@
class RnnStack : public AStack
{
const size_t NUM_CODES = 37;
public:
RnnStack(const std::string &name, size_t numContext);
RnnStack(const RnnStack& orig) = delete;
@@ -34,9 +36,20 @@ public:
arma::mat vc_to_v(const arma::mat &vc) const;
arma::mat vc_to_c(const arma::mat &vc) const;
arma::mat step_forward(arma::mat &state, const arma::mat &v);
arma::mat to_next(const arma::mat &v)
{
return v.cols(0, NUM_CODES-1);
}
arma::mat to_curr(const arma::mat &v)
{
return v.cols(NUM_CODES, 2*NUM_CODES-1);
}
private:
size_t m_numContext;
};
#endif /* RNNSTACK_HPP */