| Stan Math Library 2.12.0 reverse mode automatic differentiation

- Main Page
- Modules
- Namespaces
- Classes
- Files

- File List
- File Members

csr_to_dense_matrix.hpp
Go to the documentation of this file.
1 #ifndef STAN_MATH_PRIM_MAT_FUN_CSR_TO_DENSE_MATRIX_HPP
2 #define STAN_MATH_PRIM_MAT_FUN_CSR_TO_DENSE_MATRIX_HPP
3
4 #include <stan/math/prim/mat/err/check_range.hpp>
5 #include <stan/math/prim/mat/fun/csr_u_to_z.hpp>
6 #include <stan/math/prim/mat/fun/Eigen.hpp>
7 #include <stan/math/prim/mat/fun/dot_product.hpp>
8 #include <stan/math/prim/scal/err/check_size_match.hpp>
9 #include <stan/math/prim/scal/err/check_positive.hpp>
10 #include <vector>
11
12 namespace stan {
13 namespace math {
14
35 template <typename T>
36 inline Eigen::Matrix<T, Eigen::Dynamic, Eigen::Dynamic>
37 csr_to_dense_matrix(int m,
38 int n,
39 const Eigen::Matrix<T, Eigen::Dynamic, 1>& w,
40 const std::vector<int>& v,
41 const std::vector<int>& u) {
42 using Eigen::Dynamic;
43 using Eigen::Matrix;
44
45 check_positive("csr_to_dense_matrix", "m", m);
46 check_positive("csr_to_dense_matrix", "n", n);
47 check_size_match("csr_to_dense_matrix", "m", m, "u", u.size()-1);
48 check_size_match("csr_to_dense_matrix", "w", w.size(), "v", v.size());
49 check_size_match("csr_to_dense_matrix", "u/z",
50 u[m-1] + csr_u_to_z(u, m - 1) - 1,
51 "v", v.size());
52 for (size_t i = 0; i < v.size(); ++i)
53 check_range("csr_to_dense_matrix", "v[]", n, v[i]);
54
55 Matrix<T, Dynamic, Dynamic> result(m, n);
56 result.setZero();
57 for (int row = 0; row < m; ++row) {
58 int row_end_in_w = (u[row] - stan::error_index::value)
59 + csr_u_to_z(u, row);
60 check_range("csr_to_dense_matrix", "w", w.size(), row_end_in_w);
61 for (int nze = u[row] - stan::error_index::value;
62 nze < row_end_in_w; ++nze) {
63 // row is row index, v[nze] is column index. w[nze] is entry value.
64 check_range("csr_to_dense_matrix", "j", n, v[nze]);
65 result(row, v[nze] - stan::error_index::value) = w(nze);
66 }
67 }
68 return result;
69 } // end of csr_format group
71
72 }
73 }
74 #endif
check_size_match.hpp
check_positive.hpp
stan
Definition: log_sum_exp.hpp:8
stan::math::csr_u_to_z
int csr_u_to_z(const std::vector< int > &u, int i)
Return the z vector computed from the specified u vector at the index for the z vector.
Definition: csr_u_to_z.hpp:26
stan::math::check_range
bool check_range(const char *function, const char *name, const int max, const int index, const int nested_level, const char *error_msg)
Return true if specified index is within range.
Definition: check_range.hpp:29
stan::math::row
Eigen::Matrix< T, 1, Eigen::Dynamic > row(const Eigen::Matrix< T, Eigen::Dynamic, Eigen::Dynamic > &m, size_t i)
Return the specified row of the specified matrix, using start-at-1 indexing.
Definition: row.hpp:26
stan::error_index::value
Definition: error_index.hpp:7
stan::math::check_positive
bool check_positive(const char *function, const char *name, const T_y &y)
Return true if y is positive.
Definition: check_positive.hpp:66
stan::math::check_size_match
bool check_size_match(const char *function, const char *name_i, T_size1 i, const char *name_j, T_size2 j)
Return true if the provided sizes match.
Definition: check_size_match.hpp:30
dot_product.hpp
Eigen.hpp
csr_u_to_z.hpp
stan::math::csr_to_dense_matrix
Eigen::Matrix< T, Eigen::Dynamic, Eigen::Dynamic > csr_to_dense_matrix(int m, int n, const Eigen::Matrix< T, Eigen::Dynamic, 1 > &w, const std::vector< int > &v, const std::vector< int > &u)
Construct a dense Eigen matrix from the CSR format components.
Definition: csr_to_dense_matrix.hpp:37
check_range.hpp

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