diff options
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) |