jens/Rbm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

- integrated RnnStack

Browse Source 
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@826 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
jens
2022-01-18 08:36:10 +00:00
parent 9b2861e3d8
commit 17770f402c
10 changed files with 130 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/RnnStack.hpp
+11 -3
View File
@@ -19,16 +19,24 @@
class RnnStack : public AStack
{
public:
RnnStack(const std::string &name);
RnnStack(const std::string &name, size_t numContext);
RnnStack(const RnnStack& orig) = delete;
virtual ~RnnStack();
arma::mat trainingBatchFrom(size_t layerId, const arma::mat& batch) override;
void train(const arma::mat& batch, Rbm::IListener* pListener) override;
void train(size_t layerId, 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;
private:
size_t m_numContext;
};
#endif /* RNNSTACK_HPP */