brainpy.math.sparse: Sparse Operators

brainpy.math.sparse: Sparse Operators#

csrmv

Product of CSR sparse matrix and a dense vector using cuSPARSE algorithm.

coomv

Product of COO sparse matrix and a dense vector.

seg_matmul

Sparse matrix multiplication.

csr_to_dense

Convert a CSR sparse matrix to a dense array.

csr_to_coo

Given CSR (indices, indptr) return COO (row, col)

coo_to_csr

Convert COO (pre_ids, post_ids) connectivity to CSR (indices, indptr).