1 files changed, 14 insertions, 0 deletions
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;