math/prim_2fun_2columns__dot__product_8hpp_source.html

#ifndef STAN_MATH_PRIM_FUN_COLUMNS_DOT_PRODUCT_HPP

#define STAN_MATH_PRIM_FUN_COLUMNS_DOT_PRODUCT_HPP


#include <stan/math/prim/err.hpp>

#include <stan/math/prim/fun/Eigen.hpp>


namespace stan {

namespace math {


template <typename Mat1, typename Mat2,

          require_all_eigen_t<Mat1, Mat2>* = nullptr,

          require_all_not_eigen_vt<is_var, Mat1, Mat2>* = nullptr>

inline Eigen::Matrix<return_type_t<Mat1, Mat2>, 1, Mat1::ColsAtCompileTime>

columns_dot_product(const Mat1& v1, const Mat2& v2) {

  check_matching_sizes("columns_dot_product", "v1", v1, "v2", v2);

  return v1.cwiseProduct(v2).colwise().sum();

}


}  // namespace math

}  // namespace stan


#endif

Eigen.hpp

stan::math::check_matching_sizes
void check_matching_sizes(const char *function, const char *name1, const T_y1 &y1, const char *name2, const T_y2 &y2)
Check if two structures at the same size.
Definition check_matching_sizes.hpp:24

stan::math::columns_dot_product
auto columns_dot_product(const T_a &a, const T_b &b)
Returns the dot product of columns of the specified matrices.
Definition columns_dot_product.hpp:27

stan
The lgamma implementation in stan-math is based on either the reentrant safe lgamma_r implementation ...
Definition unit_vector_constrain.hpp:15

err.hpp