- numcontext is property of Layer

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@844 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
2022-01-19 16:05:23 +00:00
parent 2f3c10095a
commit 6a05a88d15
7 changed files with 60 additions and 80 deletions
+2 -6
View File
@@ -21,18 +21,15 @@ class RnnStack : public AStack
const size_t NUM_CODES = 37;
public:
RnnStack(const std::string &name, size_t numContext);
RnnStack(const std::string &name);
RnnStack(const RnnStack& orig) = delete;
virtual ~RnnStack();
void train(const arma::mat& batch, Rbm::IListener* pListener) override;
size_t getSeqLen();
size_t numContext();
arma::mat v_to_vc(const arma::mat &v) const;
arma::mat v_to_vc(const arma::mat &v, const arma::mat &c) const;
arma::mat vc_to_v(const arma::mat &vc) const;
arma::mat vc_to_c(const arma::mat &vc) const;
arma::mat vc_to_v(const arma::mat &vc, size_t numContext) const;
arma::mat step_forward(arma::mat &state, const arma::mat &v);
@@ -76,7 +73,6 @@ public:
private:
size_t m_numContext;
};
#endif /* RNNSTACK_HPP */