elt_divide.hpp

Go to the documentation of this file.

#ifndef STAN_MATH_PRIM_FUN_ELT_DIVIDE_HPP
#define STAN_MATH_PRIM_FUN_ELT_DIVIDE_HPP
 
#include <stan/math/prim/meta.hpp>
#include <stan/math/prim/err.hpp>
#include <stan/math/prim/fun/Eigen.hpp>
#include <stan/math/prim/fun/divide.hpp>
 
namespace stan {
namespace math {
 
template <typename Mat1, typename Mat2,
          require_all_eigen_t<Mat1, Mat2>* = nullptr,
          require_all_not_st_var<Mat1, Mat2>* = nullptr>
inline auto elt_divide(Mat1&& m1, Mat2&& m2) {
  check_matching_dims("elt_divide", "m1", m1, "m2", m2);
  return make_holder(
      [&](auto&& m1_, auto&& m2_) {
        return (m1_.array() / m2_.array()).matrix();
      },
      std::forward<Mat1>(m1), std::forward<Mat2>(m2));
}
 
template <typename Mat, typename Scal, require_matrix_t<Mat>* = nullptr,
          require_stan_scalar_t<Scal>* = nullptr>
inline auto elt_divide(Mat&& m, Scal s) {
  return divide(std::forward<Mat>(m), s);
}
 
template <typename Scal, typename Mat, require_stan_scalar_t<Scal>* = nullptr,
          require_eigen_t<Mat>* = nullptr>
inline auto elt_divide(Scal s, Mat&& m) {
  return make_holder([s](auto&& m_) { return (s / m_.array()).matrix(); },
                     std::forward<Mat>(m));
}
 
template <typename Scal1, typename Scal2,
          require_all_stan_scalar_t<Scal1, Scal2>* = nullptr>
inline auto elt_divide(Scal1 s1, Scal2 s2) {
  return s1 / s2;
}
 
}  // namespace math
}  // namespace stan
 
#endif