take into account non-square matrices

conradsnicta · Aug 2, 2017 · 763ed1a · 763ed1a
1 parent 82bf6a8
commit 763ed1a
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/include/armadillo_bits/SpMat_meat.hpp b/include/armadillo_bits/SpMat_meat.hpp
@@ -3392,6 +3392,12 @@ SpMat<eT>::eye(const uword in_rows, const uword in_cols)
 
   for(uword i = 0; i <= N; ++i) { access::rw(col_ptrs[i])    = i; }
 
+  // take into account non-square matrices
+  for(uword i = (N+1); i <= in_cols; ++i)
+    {
+    access::rw(col_ptrs[i]) += col_ptrs[i - 1];
+    }
+
   access::rw(n_nonzero) = N;
 
   return *this;