- added up_pass() and down_pass()

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@580 b431acfa-c32f-4a4a-93f1-934dc6c82436

source/Layer.cpp

@@ -102,12 +102,22 @@ Json::Value Layer::toJson() const
 
 }
 
-void Layer::up_pass(const arma::mat& v)
+arma::mat Layer::up_pass(const arma::mat &hidden)
 {
-
+	arma::mat reconstruction = rbm().toVisibleProbs(hidden);
+	if (upper)
+	{
+		return upper->up_pass(reconstruction);
+	}
+	return rbm().toHiddenProbs(reconstruction);
 }
 
-void Layer::down_pass(arma::mat& dst, const arma::mat& src)
+arma::mat Layer::down_pass(const arma::mat &visible)
 {
-
+	arma::mat hidden = rbm().toHiddenProbs(visible);
+	if (lower)
+	{
+		return lower->down_pass(hidden);
+	}
+	return rbm().toVisibleProbs(hidden);
 }

source/Layer.hpp

@@ -35,8 +35,8 @@ public:
 	Rbm& rbm();
 	void saveWeights();
 
-	void up_pass(const arma::mat& v);
-	void down_pass(arma::mat& dst, const arma::mat& src);
+	arma::mat up_pass(const arma::mat& hidden);
+	arma::mat down_pass(const arma::mat& visible);
 
 	size_t id()
 	{