diff options
| author | Qi Liu <liuq901@163.com> | 2016-04-29 16:00:20 +0800 |
|---|---|---|
| committer | Qi Liu <liuq901@163.com> | 2016-04-29 16:00:20 +0800 |
| commit | 1f5568a9e1457dcf5aadd08749aef6194370b43f (patch) | |
| tree | d705bbec5b701774a144810c5e60a73e346f9c39 /nerv/matrix/generic/mmatrix.c | |
| parent | 6051c37a2c55365a5834d5b4c11d973e2e4e1ad4 (diff) | |
add relu layer
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)}, |