From df8c51ab1f7153da6c91e29d9ff38af7864c5881 Mon Sep 17 00:00:00 2001
From: Joe Zhao <ztuowen@gmail.com>
Date: Wed, 8 Oct 2014 17:16:24 +0800
Subject: add getin & get out++++
---
src/CuTNetLib/cuMisc.h | 14 ++++++++++++++
1 file changed, 14 insertions(+)
diff --git a/src/CuTNetLib/cuMisc.h b/src/CuTNetLib/cuMisc.h
index 207b210..5fbcf9d 100644
--- a/src/CuTNetLib/cuMisc.h
+++ b/src/CuTNetLib/cuMisc.h
@@ -331,6 +331,13 @@ namespace TNet {
ErrorInputVec[pos]=&rErrorInput;
}
+ void CuMatrix<BaseFloat>& GetOutput(int pos=0)
+ {
+ if (pos>=0 && pos<size)
+ return *OutputVec[pos];
+ return *OutputVec[0];
+ }
+
void Propagate()
{
if (NULL == mpInput) Error("mpInput is NULL");
@@ -434,6 +441,13 @@ namespace TNet {
InputVec[pos]=&rInput;
}
+ const CuMatrix<BaseFloat>& GetErrorOutput(int pos=0)
+ {
+ if (pos>=0 && pos<size)
+ return *ErrorOutputVec[i];
+ return *ErrorOutputVec[0];
+ }
+
void Propagate()
{
int loc=0;
--
cgit v1.2.3-70-g09d2