update diagonlal lstm

1 files changed, 17 insertions, 0 deletions

diff --git a/nerv/lib/matrix/generic/mmatrix.c b/nerv/lib/matrix/generic/mmatrix.c
index fb99b53..6272cbe 100644
--- a/nerv/lib/matrix/generic/mmatrix.c
+++ b/nerv/lib/matrix/generic/mmatrix.c
@@ -274,6 +274,23 @@ void nerv_matrix_(fill)(Matrix *self, double val,
     NERV_SET_STATUS(status, NERV_NORMAL, 0);
 }
 
+void nerv_matrix_(diagonalize)(Matrix *selfa,
+                                MContext *context, Status *status) {
+    if (self->nrow != self->ncol)
+        NERV_EXIT_STATUS(status, MAT_MISMATCH_DIM, 0);
+    int i, j;
+    size_t astride = self->stride;
+    MATRIX_ELEM *arow = MATRIX_ELEM_PTR(self);
+    for (i = 0; i < self->nrow; i++)
+    {
+        for (j = 0; j < self->ncol; j++)
+            if (i != j)
+                arow[j] = 0;
+        arow = MATRIX_NEXT_ROW_PTR(arow, astride);
+    }
+    NERV_SET_STATUS(status, NERV_NORMAL, 0);
+}
+
 void nerv_matrix_(sigmoid)(Matrix *a, const Matrix *b,
                             MContext *context, Status *status) {
     CHECK_SAME_DIMENSION(a, b, status);