initialize.hpp

Go to the documentation of this file.

#ifndef STAN_MATH_PRIM_FUN_INITIALIZE_HPP
#define STAN_MATH_PRIM_FUN_INITIALIZE_HPP
 
#include <stan/math/prim/fun/Eigen.hpp>
#include <stan/math/prim/meta.hpp>
#include <vector>
#include <type_traits>
 
namespace stan {
namespace math {
 
// initializations called for local variables generate in Stan
// code; fills in all cells in first arg with second arg
 
template <typename T, typename V, require_all_stan_scalar_t<T, V>* = nullptr,
          require_convertible_t<V, T>* = nullptr>
inline void initialize(T& x, V v) {
  x = v;
}
 
template <typename T, int R, int C, typename V>
inline void initialize(Eigen::Matrix<T, R, C>& x, const V& v) {
  x.fill(v);
}
 
template <typename T, typename V>
inline void initialize(std::vector<T>& x, const V& v) {
  for (size_t i = 0; i < x.size(); ++i) {
    initialize(x[i], v);
  }
}
 
}  // namespace math
}  // namespace stan
#endif