diff options
author | Qi Liu <liuq901@163.com> | 2016-02-24 16:16:03 +0800 |
---|---|---|
committer | Qi Liu <liuq901@163.com> | 2016-02-24 16:16:03 +0800 |
commit | a51498d2761714f4e034f036b84b4b89a88e9539 (patch) | |
tree | ff78fabd169c3c0346453c7005c84b176ac49ca6 /nerv/matrix/generic/matrix.c | |
parent | 9642bd16922b288c81dee25f17373466ae6888c4 (diff) |
update LSTM layer
Diffstat (limited to 'nerv/matrix/generic/matrix.c')
-rw-r--r-- | nerv/matrix/generic/matrix.c | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/nerv/matrix/generic/matrix.c b/nerv/matrix/generic/matrix.c index c1da774..3162ffb 100644 --- a/nerv/matrix/generic/matrix.c +++ b/nerv/matrix/generic/matrix.c @@ -338,4 +338,12 @@ static int nerv_matrix_(lua_scale_rows_by_row)(lua_State *L) { return 0; } +static int nerv_matrix_(lua_diagonalize)(lua_State *L) { + Status status; + Matrix *a = luaT_checkudata(L, 1, nerv_matrix_(tname)); + nerv_matrix_(diagonalize)(a, &status); + NERV_LUA_CHECK_STATUS(L, status); + return 0; +} + #endif |