From 497611e62608564561fb1e123725281991ae05f0 Mon Sep 17 00:00:00 2001 From: Joe Zhao Date: Wed, 8 Oct 2014 19:16:16 +0800 Subject: Reorder --- src/CuTNetLib/cuMisc.h | 4 ++++ 1 file changed, 4 insertions(+) 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& 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& rErrorInput,int pos=0) { + if (pos == 0) + mpErrorInput = &rErrorInput; PipeVec[Order[pos]]->SetErrorInput(rErrorInput); } -- cgit v1.2.3-70-g09d2