diff options
| author | mfy43 <ted.sybil@gmail.com> | 2016-04-30 14:48:29 +0800 |
|---|---|---|
| committer | mfy43 <ted.sybil@gmail.com> | 2016-04-30 14:48:29 +0800 |
| commit | 1aaffa6b9a56d4d580dea7bf3f1b7df1eec5da2f (patch) | |
| tree | d705bbec5b701774a144810c5e60a73e346f9c39 /nerv/test/matrix_func.lua | |
| parent | 6051c37a2c55365a5834d5b4c11d973e2e4e1ad4 (diff) | |
| parent | 1f5568a9e1457dcf5aadd08749aef6194370b43f (diff) | |
Merge branch 'master' into 'master'
add relu layer
See merge request !2
Diffstat (limited to 'nerv/test/matrix_func.lua')
| -rw-r--r-- | nerv/test/matrix_func.lua | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/nerv/test/matrix_func.lua b/nerv/test/matrix_func.lua index 817d463..90bb27f 100644 --- a/nerv/test/matrix_func.lua +++ b/nerv/test/matrix_func.lua @@ -164,6 +164,9 @@ function _test_all_shape(mat_type, m, n, k, fill) local c = a:create() c:tanh(a) print(c) + a:add(a, c, 1.0, -3.0) + c:relu(a) + print(c) end function test_all(mat_type) _test_all_shape(mat_type, 3, 4, 2, _pattern_fill) |