diff options
| -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]; } |