From a51498d2761714f4e034f036b84b4b89a88e9539 Mon Sep 17 00:00:00 2001
From: Qi Liu <liuq901@163.com>
Date: Wed, 24 Feb 2016 16:16:03 +0800
Subject: update LSTM layer

---
 nerv/matrix/generic/matrix.c | 8 ++++++++
 1 file changed, 8 insertions(+)

(limited to 'nerv/matrix/generic/matrix.c')

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
-- 
cgit v1.2.3