Automatic Differentiation
 
Loading...
Searching...
No Matches
poisson_binomial_lpmf.hpp
Go to the documentation of this file.
1#ifndef STAN_MATH_PRIM_PROB_POISSON_BINOMIAL_LPMF_HPP
2#define STAN_MATH_PRIM_PROB_POISSON_BINOMIAL_LPMF_HPP
3
10
11namespace stan {
12namespace math {
13
27template <bool propto, typename T_y, typename T_theta>
29 const T_theta& theta) {
30 static constexpr const char* function = "poisson_binomial_lpmf";
31
32 auto size_theta = size_mvt(theta);
33 if (size_theta > 1) {
34 check_consistent_sizes(function, "Successes variables", y,
35 "Probability parameters", theta);
36 }
37
38 auto max_sz = std::max(stan::math::size(y), size_theta);
39 scalar_seq_view<T_y> y_vec(y);
40 vector_seq_view<T_theta> theta_vec(theta);
41
42 for (size_t i = 0; i < max_sz; ++i) {
43 check_bounded(function, "Successes variable", y_vec[i], 0,
44 theta_vec[i].size());
45 check_finite(function, "Probability parameters", theta_vec.val(i));
46 check_bounded(function, "Probability parameters", theta_vec.val(i), 0.0,
47 1.0);
48 }
49
50 return_type_t<T_theta> log_prob = 0.0;
51 for (size_t i = 0; i < max_sz; ++i) {
52 auto x = poisson_binomial_log_probs(y_vec[i], theta_vec[i]);
53 log_prob += x(y_vec[i]);
54 }
55
56 return log_prob;
57}
58
59template <typename T_y, typename T_theta>
61 const T_theta& theta) {
62 return poisson_binomial_lpmf<false>(y, theta);
63}
64
65} // namespace math
66} // namespace stan
67#endif
scalar_seq_view provides a uniform sequence-like wrapper around either a scalar or a sequence of scal...
This class provides a low-cost wrapper for situations where you either need an Eigen Vector or RowVec...
return_type_t< T_theta > poisson_binomial_lpmf(const T_y &y, const T_theta &theta)
Returns the log PMF for the Poisson-binomial distribution evaluated at an specified array of numbers ...
int64_t size_mvt(const ScalarT &)
Provides the size of a multivariate argument.
Definition size_mvt.hpp:25
typename return_type< Ts... >::type return_type_t
Convenience type for the return type of the specified template parameters.
int64_t size(const T &m)
Returns the size (number of the elements) of a matrix_cl or var_value<matrix_cl<T>>.
Definition size.hpp:19
plain_type_t< T_theta > poisson_binomial_log_probs(int y, const T_theta &theta)
Returns the last row of the log probability matrix of the Poisson-Binomial distribution given the num...
void check_bounded(const char *function, const char *name, const T_y &y, const T_low &low, const T_high &high)
Check if the value is between the low and high values, inclusively.
void check_consistent_sizes(const char *)
Trivial no input case, this function is a no-op.
void check_finite(const char *function, const char *name, const T_y &y)
Return true if all values in y are finite.
The lgamma implementation in stan-math is based on either the reentrant safe lgamma_r implementation ...