diff options
| author | Joe Zhao <ztuowen@gmail.com> | 2014-10-08 17:18:44 +0800 |
|---|---|---|
| committer | Joe Zhao <ztuowen@gmail.com> | 2014-10-08 17:18:44 +0800 |
| commit | 3eed0cecd953ce40f9d541f0af86f32f9f46aedb (patch) | |
| tree | 18eb32454446b978a44f808bbe2b00af33d17cbe /src | |
| parent | df8c51ab1f7153da6c91e29d9ff38af7864c5881 (diff) | |
| download | tnet-3eed0cecd953ce40f9d541f0af86f32f9f46aedb.tar.gz tnet-3eed0cecd953ce40f9d541f0af86f32f9f46aedb.tar.bz2 tnet-3eed0cecd953ce40f9d541f0af86f32f9f46aedb.zip | |
add getin & get out+++++
Diffstat (limited to 'src')
| -rw-r--r-- | src/CuTNetLib/cuMisc.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/CuTNetLib/cuMisc.h b/src/CuTNetLib/cuMisc.h index 5fbcf9d..137dbcf 100644 --- a/src/CuTNetLib/cuMisc.h +++ b/src/CuTNetLib/cuMisc.h @@ -331,7 +331,7 @@ namespace TNet { ErrorInputVec[pos]=&rErrorInput; } - void CuMatrix<BaseFloat>& GetOutput(int pos=0) + const CuMatrix<BaseFloat>& GetOutput(int pos=0) { if (pos>=0 && pos<size) return *OutputVec[pos]; @@ -444,7 +444,7 @@ namespace TNet { const CuMatrix<BaseFloat>& GetErrorOutput(int pos=0) { if (pos>=0 && pos<size) - return *ErrorOutputVec[i]; + return *ErrorOutputVec[pos]; return *ErrorOutputVec[0]; } |