| Stan Math Library 2.12.0 reverse mode automatic differentiation

- Main Page
- Modules
- Namespaces
- Classes
- Files

- File List
- File Members

csr_extract_u.hpp
Go to the documentation of this file.
1 #ifndef STAN_MATH_PRIM_MAT_FUN_CSR_EXTRACT_U_HPP
2 #define STAN_MATH_PRIM_MAT_FUN_CSR_EXTRACT_U_HPP
3
4 #include <stan/math/prim/scal/meta/error_index.hpp>
5 #include <stan/math/prim/mat/fun/Eigen.hpp>
6 #include <Eigen/Sparse>
7 #include <vector>
8 #include <numeric>
9
10 namespace stan {
11 namespace math {
12
24 template <typename T>
25 const std::vector<int>
26 csr_extract_u(const Eigen::SparseMatrix<T, Eigen::RowMajor>& A) {
27 std::vector<int> u(A.outerSize() + 1); // last entry is garbage.
28 for (int nze = 0; nze <= A.outerSize(); ++nze)
29 u[nze] = *(A.outerIndexPtr() + nze) + stan::error_index::value;
30 return u;
31 }
32
40 template <typename T, int R, int C>
41 const std::vector<int>
42 csr_extract_u(const Eigen::Matrix<T, R, C>& A) {
43 Eigen::SparseMatrix<T, Eigen::RowMajor> B = A.sparseView();
44 std::vector<int> u = csr_extract_u(B);
45 return u;
46 }
47 // end of csr_format group
49
50 }
51 }
52
53 #endif
stan
Definition: log_sum_exp.hpp:8
stan::error_index::value
Definition: error_index.hpp:7
Eigen.hpp
error_index.hpp
stan::math::csr_extract_u
const std::vector< int > csr_extract_u(const Eigen::SparseMatrix< T, Eigen::RowMajor > &A)
Extract the NZE index for each entry from a sparse matrix.
Definition: csr_extract_u.hpp:26

[ Stan Home Page ] © 2011–2016, Stan Development Team.