pub fn compute_nnz_per_row(a: &CsrMatrix) -> Vec<usize>
Compute the number of non-zeros per row for a CSR matrix.