added function to get sparse matrix

JuliaSNN · Nov 28, 2024 · 20f4889 · 20f4889
1 parent 1164779
commit 20f4889
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/util.jl b/src/util.jl
@@ -1,10 +1,14 @@
 function connect!(c, j, i, μ = 1e-6)
-    W = sparse(c.I, c.J, c.W, length(c.rowptr) - 1, length(c.colptr) - 1)
+    W = matrix(c)
     W[i, j] = μ * randn(Float32)
     replace_sparse_matrix!(c, W)
     return nothing
 end
 
+function matrix(c)
+    return sparse(c.I, c.J, c.W, length(c.rowptr) - 1, length(c.colptr) - 1)
+end
+
 function replace_sparse_matrix!(c::S, W::SparseMatrixCSC) where S <: AbstractConnection
     rowptr, colptr, I, J, index, W = dsparse(W)
     @assert length(rowptr) == length(c.rowptr) "Rowptr length mismatch"