1 files changed, 4 insertions, 0 deletions
diff --git a/src/CuTNetLib/cuMisc.h b/src/CuTNetLib/cuMisc.h
index 3447739..4d25a05 100644
--- a/src/CuTNetLib/cuMisc.h
+++ b/src/CuTNetLib/cuMisc.h
@@ -581,11 +581,15 @@ namespace TNet {
     /// Set input vector (bind with the preceding NetworkComponent)
     void SetInput(const CuMatrix<BaseFloat>& rInput,int pos=0)
     {
+      if (pos == 0)
+        mpInput = &rInput;
       PipeVec[pos]->SetInput(rInput);
     }          
     /// Set error input vector (bind with the following NetworkComponent) 
     void SetErrorInput(const CuMatrix<BaseFloat>& rErrorInput,int pos=0)
     {
+      if (pos == 0)
+        mpErrorInput = &rErrorInput;
       PipeVec[Order[pos]]->SetErrorInput(rErrorInput);
     }