1 #ifndef STAN_MATH_PRIM_MAT_FUN_QUAD_FORM_DIAG_HPP 2 #define STAN_MATH_PRIM_MAT_FUN_QUAD_FORM_DIAG_HPP 5 #include <boost/math/tools/promotion.hpp> 13 template <
typename T1,
typename T2,
int R,
int C>
14 inline Eigen::Matrix <
15 typename boost::math::tools::promote_args<T1, T2>::type,
16 Eigen::Dynamic, Eigen::Dynamic>
18 const Eigen::Matrix<T2, R, C>& vec) {
19 using boost::math::tools::promote_args;
22 int size = vec.size();
25 Eigen::Matrix<typename promote_args<T1, T2>::type,
26 Eigen::Dynamic, Eigen::Dynamic> result(size, size);
27 for (
int i = 0; i <
size; i++) {
28 result(i, i) = vec(i)*vec(i)*mat(i, i);
29 for (
int j = i+1; j <
size; ++j) {
30 typename promote_args<T1, T2>::type temp = vec(i)*vec(j);
31 result(j, i) = temp*mat(j, i);
32 result(i, j) = temp*mat(i, j);
void check_vector(const char *function, const char *name, const Eigen::Matrix< T, R, C > &x)
Check if the matrix is either a row vector or column vector.
void check_size_match(const char *function, const char *name_i, T_size1 i, const char *name_j, T_size2 j)
Check if the provided sizes match.
void check_square(const char *function, const char *name, const Eigen::Matrix< T_y, Eigen::Dynamic, Eigen::Dynamic > &y)
Check if the specified matrix is square.
int size(const std::vector< T > &x)
Return the size of the specified standard vector.
Eigen::Matrix< typename boost::math::tools::promote_args< T1, T2 >::type, Eigen::Dynamic, Eigen::Dynamic > quad_form_diag(const Eigen::Matrix< T1, Eigen::Dynamic, Eigen::Dynamic > &mat, const Eigen::Matrix< T2, R, C > &vec)