aboutsummaryrefslogtreecommitdiff
path: root/matrix/generic
diff options
context:
space:
mode:
authorDeterminant <[email protected]>2015-05-28 11:28:24 +0800
committerDeterminant <[email protected]>2015-05-28 11:28:24 +0800
commit51ee28426b992b1421ded46721ade7a57b0a1896 (patch)
treea877f836c428df264ae20ab8c876eabf36c4a71c /matrix/generic
parentc2abcd032e9f64bdbf015567d76edfada55c8866 (diff)
sigmoid layer is now working; clean up Makefile
Diffstat (limited to 'matrix/generic')
-rw-r--r--matrix/generic/cumatrix.c5
1 files changed, 2 insertions, 3 deletions
diff --git a/matrix/generic/cumatrix.c b/matrix/generic/cumatrix.c
index 2b3b9d4..f24280c 100644
--- a/matrix/generic/cumatrix.c
+++ b/matrix/generic/cumatrix.c
@@ -98,9 +98,8 @@ static int nerv_matrix_(sigmoid)(lua_State *L) {
Matrix *a = luaT_checkudata(L, 1, nerv_matrix_(tname));
Matrix *b = luaT_checkudata(L, 2, nerv_matrix_(tname));
CHECK_SAME_DIMENSION(a, b);
- cudak_(cuda_sigmoid)(a, b);
- luaT_pushudata(L, b, nerv_matrix_(tname));
- return 1;
+ cudak_(cuda_sigmoid)(b, a);
+ return 0;
}
static int nerv_matrix_(sigmoid_grad)(lua_State *L) {