diff options
author | Determinant <ted.sybil@gmail.com> | 2016-05-01 00:44:43 +0800 |
---|---|---|
committer | Determinant <ted.sybil@gmail.com> | 2016-05-01 00:44:43 +0800 |
commit | 34fe366898bc48b4e1ad6fa945dad8821857a459 (patch) | |
tree | 1210e1a28dd4c0f6e1cefe0b98c95b333ceec158 /nerv/matrix/generic/mmatrix.c | |
parent | 2bb64ecc5e0350f89cff1e978a1b8dcca1528a06 (diff) | |
parent | 1aaffa6b9a56d4d580dea7bf3f1b7df1eec5da2f (diff) |
Merge branch 'master' of ssh://speechlab.sjtu.edu.cn:8022/nerv-dev/nerv
Diffstat (limited to 'nerv/matrix/generic/mmatrix.c')
-rw-r--r-- | nerv/matrix/generic/mmatrix.c | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/nerv/matrix/generic/mmatrix.c b/nerv/matrix/generic/mmatrix.c index c03aee4..530888b 100644 --- a/nerv/matrix/generic/mmatrix.c +++ b/nerv/matrix/generic/mmatrix.c @@ -122,6 +122,8 @@ static const luaL_Reg nerv_matrix_(extra_methods)[] = { {"sigmoid_grad", nerv_matrix_(lua_sigmoid_grad)}, {"tanh", nerv_matrix_(lua_tanh)}, {"tanh_grad", nerv_matrix_(lua_tanh_grad)}, + {"relu", nerv_matrix_(lua_relu)}, + {"relu_grad", nerv_matrix_(lua_relu_grad)}, {"softmax", nerv_matrix_(lua_softmax)}, {"mul_elem", nerv_matrix_(lua_mul_elem)}, {"log_elem", nerv_matrix_(lua_log_elem)}, |