From 0463a64a93a6059b010abd093343f2f6400d68ca Mon Sep 17 00:00:00 2001
From: Jens Ahrensfeld <jens@jayfield.org>
Date: Fri, 25 Oct 2019 17:56:43 +0000
Subject: [PATCH] - 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 | 18 ++++++++++++++----
 source/Layer.hpp |  4 ++--
 2 files changed, 16 insertions(+), 6 deletions(-)

diff --git a/source/Layer.cpp b/source/Layer.cpp
index 695850c..5271824 100644
--- a/source/Layer.cpp
+++ b/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);
 }
diff --git a/source/Layer.hpp b/source/Layer.hpp
index b6607e5..287bbdd 100644
--- a/source/Layer.hpp
+++ b/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()
 	{