1 #ifndef STAN_MATH_PRIM_SCAL_PROB_POISSON_LOG_LPMF_HPP 2 #define STAN_MATH_PRIM_SCAL_PROB_POISSON_LOG_LPMF_HPP 19 template <
bool propto,
typename T_n,
typename T_log_rate>
21 const T_n& n,
const T_log_rate& alpha) {
25 static const char*
function =
"poisson_log_lpmf";
32 T_partials_return logp(0.0);
47 for (
size_t i = 0; i <
size; i++)
48 if (std::numeric_limits<double>::infinity() == alpha_vec[i])
50 for (
size_t i = 0; i <
size; i++)
51 if (-std::numeric_limits<double>::infinity() == alpha_vec[i]
61 for (
size_t i = 0; i <
length(alpha); i++)
65 for (
size_t i = 0; i <
size; i++) {
66 if (!(alpha_vec[i] == -std::numeric_limits<double>::infinity()
69 logp -=
lgamma(n_vec[i] + 1.0);
71 logp += n_vec[i] *
value_of(alpha_vec[i]) - exp_alpha[i];
75 ops_partials.
edge1_.partials_[i] += n_vec[i] - exp_alpha[i];
77 return ops_partials.
build(logp);
80 template <
typename T_n,
typename T_log_rate>
82 const T_n& n,
const T_log_rate& alpha) {
83 return poisson_log_lpmf<false>(n, alpha);
boost::math::tools::promote_args< double, typename partials_type< typename scalar_type< T >::type >::type, typename partials_return_type< T_pack... >::type >::type type
fvar< T > lgamma(const fvar< T > &x)
Return the natural logarithm of the gamma function applied to the specified argument.
T value_of(const fvar< T > &v)
Return the value of the specified variable.
Extends std::true_type when instantiated with zero or more template parameters, all of which extend t...
scalar_seq_view provides a uniform sequence-like wrapper around either a scalar or a sequence of scal...
This template builds partial derivatives with respect to a set of operands.
size_t length(const std::vector< T > &x)
Returns the length of the provided std::vector.
bool size_zero(T &x)
Returns 1 if input is of length 0, returns 0 otherwise.
Template metaprogram to calculate whether a summand needs to be included in a proportional (log) prob...
void check_nonnegative(const char *function, const char *name, const T_y &y)
Check if y is non-negative.
boost::math::tools::promote_args< double, typename scalar_type< T >::type, typename return_type< Types_pack... >::type >::type type
fvar< T > exp(const fvar< T > &x)
void check_not_nan(const char *function, const char *name, const T_y &y)
Check if y is not NaN.
return_type< T_log_rate >::type poisson_log_lpmf(const T_n &n, const T_log_rate &alpha)
size_t max_size(const T1 &x1, const T2 &x2)
T_return_type build(double value)
Build the node to be stored on the autodiff graph.
VectorBuilder allocates type T1 values to be used as intermediate values.
int size(const std::vector< T > &x)
Return the size of the specified standard vector.
void check_consistent_sizes(const char *function, const char *name1, const T1 &x1, const char *name2, const T2 &x2)
Check if the dimension of x1 is consistent with x2.
internal::ops_partials_edge< double, Op1 > edge1_