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 auto columns_dot_product(Mat1&& v1, Mat2&& v2) {

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

  return make_holder(

      [](auto&& v1_, auto&& v2_) {

        return v1_.cwiseProduct(v2_).colwise().sum();

      },

      to_ref(std::forward<Mat1>(v1)), to_ref(std::forward<Mat2>(v2)));

}


}  // namespace math

}  // namespace stan


#endif

Eigen.hpp

stan::math::check_matching_dims
void check_matching_dims(const char *function, const char *name1, const T1 &y1, const char *name2, const T2 &y2)
Check if the two containers have the same dimensions.
Definition check_matching_dims.hpp:29

stan::math::make_holder
auto make_holder(F &&func, Args &&... args)
Calls given function with given arguments.
Definition holder.hpp:481

stan::math::sum
auto sum(const std::vector< T > &m)
Return the sum of the entries of the specified standard vector.
Definition sum.hpp:23

stan::math::to_ref
ref_type_t< T && > to_ref(T &&a)
This evaluates expensive Eigen expressions.
Definition to_ref.hpp:18

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