sparse eye init

cai4cai · Jun 21, 2023 · de5764c · de5764c
1 parent 517a9a7
commit de5764c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/torchsparsegradutils/utils/__init__.py b/torchsparsegradutils/utils/__init__.py
@@ -8,6 +8,7 @@
     sparse_block_diag,
     sparse_block_diag_split,
     stack_csr,
+    sparse_eye,
 )
 from .random_sparse import rand_sparse, rand_sparse_tri
 
@@ -26,4 +27,5 @@
     "rand_sparse_tri",
     "sparse_block_diag_split",
     "stack_csr",
+    "sparse_eye",
 ]