Stan Math Library  2.20.0
reverse mode automatic differentiation
chi_square_rng.hpp
Go to the documentation of this file.
1 #ifndef STAN_MATH_PRIM_SCAL_PROB_CHI_SQUARE_RNG_HPP
2 #define STAN_MATH_PRIM_SCAL_PROB_CHI_SQUARE_RNG_HPP
3 
8 #include <boost/random/chi_squared_distribution.hpp>
9 #include <boost/random/variate_generator.hpp>
10 
11 namespace stan {
12 namespace math {
13 
27 template <typename T_deg, class RNG>
29  const T_deg& nu, RNG& rng) {
30  using boost::random::chi_squared_distribution;
31  using boost::variate_generator;
32 
33  static const char* function = "chi_square_rng";
34 
35  check_positive_finite(function, "Degrees of freedom parameter", nu);
36 
37  scalar_seq_view<T_deg> nu_vec(nu);
38  size_t N = length(nu);
40 
41  for (size_t n = 0; n < N; ++n) {
42  variate_generator<RNG&, chi_squared_distribution<> > chi_square_rng(
43  rng, chi_squared_distribution<>(nu_vec[n]));
44  output[n] = chi_square_rng();
45  }
46 
47  return output.data();
48 }
49 
50 } // namespace math
51 } // namespace stan
52 #endif
scalar_seq_view provides a uniform sequence-like wrapper around either a scalar or a sequence of scal...
size_t length(const std::vector< T > &x)
Returns the length of the provided std::vector.
Definition: length.hpp:16
void check_positive_finite(const char *function, const char *name, const T_y &y)
Check if y is positive and finite.
VectorBuilder allocates type T1 values to be used as intermediate values.
VectorBuilder< true, double, T_deg >::type chi_square_rng(const T_deg &nu, RNG &rng)
Return a chi squared random variate with nu degrees of freedom using the specified random number gene...

     [ Stan Home Page ] © 2011–2018, Stan Development Team.