- integrated RnnStack

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@826 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
2022-01-18 08:36:10 +00:00
parent 9b2861e3d8
commit 17770f402c
10 changed files with 130 additions and 34 deletions
+5 -1
View File
@@ -54,6 +54,10 @@ public:
StackType type();
void setName(const std::string &name);
size_t numLayers();
virtual size_t numContext()
{
return 0;
}
void addLayer(Layer *pLayer);
void delLayer(Layer *pLayer);
@@ -67,7 +71,7 @@ public:
virtual void train(size_t layerId, const arma::mat& batch, Rbm::IListener* pListener) = 0;
virtual void train(const arma::mat& batch, Rbm::IListener* pListener) = 0;
arma::mat trainingBatchFrom(size_t layerId, const arma::mat& batch);
virtual arma::mat trainingBatchFrom(size_t layerId, const arma::mat& batch) = 0;
size_t numTraining();
void addTraining(const arma::mat &toAdd);