| /dports/math/py-pystan/pystan-2.19.0.0/pystan/stan/lib/stan_math/stan/math/prim/scal/prob/ |
| H A D | neg_binomial_lpmf.hpp | 60 scalar_seq_view<T_shape> alpha_vec(alpha); in neg_binomial_lpmf() local
69 lambda[i] = value_of(alpha_vec[i]) / value_of(beta_vec[i]); in neg_binomial_lpmf()
84 = value_of(alpha_vec[i]) in neg_binomial_lpmf()
91 digamma_alpha[i] = digamma(value_of(alpha_vec[i])); in neg_binomial_lpmf()
109 - (value_of(alpha_vec[i]) / (1.0 + value_of(beta_vec[i]))); in neg_binomial_lpmf()
113 if (alpha_vec[i] > 1e10) { // reduces numerically to Poisson in neg_binomial_lpmf()
121 += n_vec[i] / value_of(alpha_vec[i]) - 1.0 / value_of(beta_vec[i]); in neg_binomial_lpmf()
129 n_vec[i] + value_of(alpha_vec[i]) - 1.0, n_vec[i]); in neg_binomial_lpmf()
135 += digamma(value_of(alpha_vec[i]) + n_vec[i]) - digamma_alpha[i] in neg_binomial_lpmf()
|
| H A D | binomial_logit_lpmf.hpp | 72 scalar_seq_view<T_prob> alpha_vec(alpha); in binomial_logit_lpmf() local
85 log_inv_logit_alpha[i] = log_inv_logit(value_of(alpha_vec[i])); in binomial_logit_lpmf()
90 log_inv_logit_neg_alpha[i] = log_inv_logit(-value_of(alpha_vec[i])); in binomial_logit_lpmf()
105 += temp1 * inv_logit(-value_of(alpha_vec[0])) in binomial_logit_lpmf()
106 - temp2 * inv_logit(value_of(alpha_vec[0])); in binomial_logit_lpmf()
112 += n_vec[i] * inv_logit(-value_of(alpha_vec[i])) in binomial_logit_lpmf()
113 - (N_vec[i] - n_vec[i]) * inv_logit(value_of(alpha_vec[i])); in binomial_logit_lpmf()
|
| H A D | poisson_log_lpmf.hpp | 53 scalar_seq_view<T_log_rate> alpha_vec(alpha); in poisson_log_lpmf() local
58 if (std::numeric_limits<double>::infinity() == alpha_vec[i]) in poisson_log_lpmf()
61 if (-std::numeric_limits<double>::infinity() == alpha_vec[i] in poisson_log_lpmf()
73 exp_alpha[i] = exp(value_of(alpha_vec[i])); in poisson_log_lpmf()
76 if (!(alpha_vec[i] == -std::numeric_limits<double>::infinity() in poisson_log_lpmf()
81 logp += n_vec[i] * value_of(alpha_vec[i]) - exp_alpha[i]; in poisson_log_lpmf()
|
| H A D | beta_rng.hpp | 48 scalar_seq_view<T_shape1> alpha_vec(alpha); in beta_rng() local
59 if (alpha_vec[n] > 1.0 && beta_vec[n] > 1.0) { in beta_rng()
61 rng, gamma_distribution<>(alpha_vec[n], 1.0)); in beta_rng()
69 rng, gamma_distribution<>(alpha_vec[n] + 1, 1.0)); in beta_rng()
72 double log_a = std::log(uniform_rng()) / alpha_vec[n] in beta_rng()
|
| H A D | beta_binomial_lpmf.hpp | 72 scalar_seq_view<T_size1> alpha_vec(alpha); in beta_binomial_lpmf() local
92 lbeta_numerator[i] = lbeta(n_vec[i] + value_of(alpha_vec[i]), in beta_binomial_lpmf()
101 = lbeta(value_of(alpha_vec[i]), value_of(beta_vec[i])); in beta_binomial_lpmf()
108 digamma_n_plus_alpha[i] = digamma(n_vec[i] + value_of(alpha_vec[i])); in beta_binomial_lpmf()
116 = digamma(N_vec[i] + value_of(alpha_vec[i]) + value_of(beta_vec[i])); in beta_binomial_lpmf()
124 = digamma(value_of(alpha_vec[i]) + value_of(beta_vec[i])); in beta_binomial_lpmf()
130 digamma_alpha[i] = digamma(value_of(alpha_vec[i])); in beta_binomial_lpmf()
|
| H A D | uniform_rng.hpp | 50 scalar_seq_view<T_alpha> alpha_vec(alpha); in uniform_rng() local
58 output[n] = (beta_vec[n] - alpha_vec[n]) * uniform_rng() + alpha_vec[n]; in uniform_rng()
|
| H A D | uniform_lpdf.hpp | 71 scalar_seq_view<T_low> alpha_vec(alpha); in uniform_lpdf() local
77 if (y_dbl < value_of(alpha_vec[n]) || y_dbl > value_of(beta_vec[n])) in uniform_lpdf()
87 = 1.0 / (value_of(beta_vec[i]) - value_of(alpha_vec[i])); in uniform_lpdf()
95 = log(value_of(beta_vec[i]) - value_of(alpha_vec[i])); in uniform_lpdf()
|
| H A D | beta_lpdf.hpp | 82 scalar_seq_view<T_scale_succ> alpha_vec(alpha); in beta_lpdf() local
117 lgamma_alpha[n] = lgamma(value_of(alpha_vec[n])); in beta_lpdf()
119 digamma_alpha[n] = digamma(value_of(alpha_vec[n])); in beta_lpdf()
146 = value_of(alpha_vec[n]) + value_of(beta_vec[n]); in beta_lpdf()
155 const T_partials_return alpha_dbl = value_of(alpha_vec[n]); in beta_lpdf()
|
| H A D | gamma_lpdf.hpp | 75 scalar_seq_view<T_shape> alpha_vec(alpha); in gamma_lpdf() local
107 lgamma_alpha[n] = lgamma(value_of(alpha_vec[n])); in gamma_lpdf()
109 digamma_alpha[n] = digamma(value_of(alpha_vec[n])); in gamma_lpdf()
122 const T_partials_return alpha_dbl = value_of(alpha_vec[n]); in gamma_lpdf()
|
| H A D | uniform_lcdf.hpp | 46 scalar_seq_view<T_low> alpha_vec(alpha); in uniform_lcdf() local
54 if (y_dbl < value_of(alpha_vec[n]) || y_dbl > value_of(beta_vec[n])) in uniform_lcdf()
62 const T_partials_return alpha_dbl = value_of(alpha_vec[n]); in uniform_lcdf()
|
| H A D | uniform_lccdf.hpp | 46 scalar_seq_view<T_low> alpha_vec(alpha); in uniform_lccdf() local
52 if (y_dbl < value_of(alpha_vec[n]) || y_dbl > value_of(beta_vec[n])) in uniform_lccdf()
61 const T_partials_return alpha_dbl = value_of(alpha_vec[n]); in uniform_lccdf()
|
| H A D | inv_gamma_lpdf.hpp | 72 scalar_seq_view<T_shape> alpha_vec(alpha); in inv_gamma_lpdf() local
107 lgamma_alpha[n] = lgamma(value_of(alpha_vec[n])); in inv_gamma_lpdf()
109 digamma_alpha[n] = digamma(value_of(alpha_vec[n])); in inv_gamma_lpdf()
121 const T_partials_return alpha_dbl = value_of(alpha_vec[n]); in inv_gamma_lpdf()
|
| /dports/math/stanmath/math-4.2.0/test/unit/math/mix/prob/ |
| H A D | dirichlet_test.cpp | 95 std::vector<Matrix<fvar<var>, Dynamic, 1>> alpha_vec(3); in TEST() local
96 alpha_vec[0] = alpha1; in TEST()
97 alpha_vec[1] = alpha2; in TEST()
98 alpha_vec[2] = alpha3; in TEST()
105 fvar<var> out = dirichlet_log(theta_vec, alpha_vec); in TEST()
123 out = dirichlet_log(theta1, alpha_vec); in TEST()
187 alpha_vec[0] = alpha1; in TEST()
188 alpha_vec[1] = alpha2; in TEST()
189 alpha_vec[2] = alpha3; in TEST()
196 fvar<fvar<var>> out = dirichlet_log(theta_vec, alpha_vec); in TEST()
[all …]
|
| /dports/math/stanmath/math-4.2.0/stan/math/prim/prob/ |
| H A D | beta_rng.hpp | 51 scalar_seq_view<T_alpha_ref> alpha_vec(alpha_ref); in beta_rng() local
62 if (alpha_vec[n] > 1.0 && beta_vec[n] > 1.0) { in beta_rng()
64 rng, gamma_distribution<>(alpha_vec[n], 1.0)); in beta_rng()
72 rng, gamma_distribution<>(alpha_vec[n] + 1, 1.0)); in beta_rng()
75 double log_a = std::log(uniform_rng()) / alpha_vec[n] in beta_rng()
|
| H A D | beta_binomial_lpmf.hpp | 79 scalar_seq_view<T_alpha_ref> alpha_vec(alpha_ref); in beta_binomial_lpmf() local
102 lbeta_denominator[i] = lbeta(alpha_vec.val(i), beta_vec.val(i)); in beta_binomial_lpmf()
108 lbeta_diff[i] = lbeta(n_vec[i] + alpha_vec.val(i), in beta_binomial_lpmf()
118 digamma_n_plus_alpha[i] = digamma(n_vec.val(i) + alpha_vec.val(i)); in beta_binomial_lpmf()
127 digamma_alpha_plus_beta[i] = digamma(alpha_vec.val(i) + beta_vec.val(i)); in beta_binomial_lpmf()
138 - digamma(N_vec.val(i) + alpha_vec.val(i) + beta_vec.val(i)); in beta_binomial_lpmf()
146 digamma_alpha[i] = digamma(alpha_vec.val(i)); in beta_binomial_lpmf()
|
| H A D | uniform_rng.hpp | 50 scalar_seq_view<T_alpha_ref> alpha_vec(alpha_ref); in uniform_rng() local
58 output[n] = (beta_vec[n] - alpha_vec[n]) * uniform_rng() + alpha_vec[n]; in uniform_rng()
|
| H A D | dirichlet_lpdf.hpp | 70 vector_seq_view<T_alpha_ref> alpha_vec(alpha_ref); in dirichlet_lpdf() local
74 "prior sample sizes", alpha_vec[0]); in dirichlet_lpdf()
77 check_positive(function, "prior sample sizes", alpha_vec[t]); in dirichlet_lpdf()
89 alpha_dbl.col(t) = alpha_vec.val(t); in dirichlet_lpdf()
|
| H A D | neg_binomial_cdf.hpp | 50 scalar_seq_view<T_alpha_ref> alpha_vec(alpha_ref); in neg_binomial_cdf() local
72 digamma_alpha_vec[i] = digamma(alpha_vec.val(i)); in neg_binomial_cdf()
76 const T_partials_return alpha_dbl = alpha_vec.val(i); in neg_binomial_cdf()
89 const T_partials_return alpha_dbl = alpha_vec.val(i); in neg_binomial_cdf()
|
| H A D | neg_binomial_lccdf.hpp | 54 scalar_seq_view<T_alpha_ref> alpha_vec(alpha_ref); in neg_binomial_lccdf() local
82 digammaAlpha_vec[i] = digamma(alpha_vec.val(i)); in neg_binomial_lccdf()
86 const T_partials_return alpha_dbl = alpha_vec.val(i); in neg_binomial_lccdf()
99 const T_partials_return alpha_dbl = alpha_vec.val(i); in neg_binomial_lccdf()
|
| H A D | beta_lccdf.hpp | 69 scalar_seq_view<T_alpha_ref> alpha_vec(alpha_ref); in beta_lccdf() local
88 digamma_alpha[i] = digamma(alpha_vec.val(i)); in beta_lccdf()
94 digamma_sum[i] = digamma(alpha_vec.val(i) + beta_vec.val(i)); in beta_lccdf()
100 const T_partials_return alpha_dbl = alpha_vec.val(n); in beta_lccdf()
|
| H A D | beta_lcdf.hpp | 69 scalar_seq_view<T_alpha_ref> alpha_vec(alpha_ref); in beta_lcdf() local
88 digamma_alpha[i] = digamma(alpha_vec.val(i)); in beta_lcdf()
94 digamma_sum[i] = digamma(alpha_vec.val(i) + beta_vec.val(i)); in beta_lcdf()
100 const T_partials_return alpha_dbl = alpha_vec.val(n); in beta_lcdf()
|
| H A D | neg_binomial_lcdf.hpp | 54 scalar_seq_view<T_alpha_ref> alpha_vec(alpha_ref); in neg_binomial_lcdf() local
82 digammaAlpha_vec[i] = digamma(alpha_vec.val(i)); in neg_binomial_lcdf()
86 const T_partials_return alpha_dbl = alpha_vec.val(i); in neg_binomial_lcdf()
99 const T_partials_return alpha_dbl = alpha_vec.val(i); in neg_binomial_lcdf()
|
| H A D | neg_binomial_lpmf.hpp | 63 scalar_seq_view<T_alpha_ref> alpha_vec(alpha_ref); in neg_binomial_lpmf() local
74 digamma_alpha[i] = digamma(alpha_vec.val(i)); in neg_binomial_lpmf()
91 const T_partials_return alpha_dbl = alpha_vec.val(i); in neg_binomial_lpmf()
99 const T_partials_return alpha_dbl = alpha_vec.val(i); in neg_binomial_lpmf()
|
| H A D | beta_cdf.hpp | 61 scalar_seq_view<T_alpha_ref> alpha_vec(alpha_ref); in beta_cdf() local
81 digamma_alpha[n] = digamma(alpha_vec.val(n)); in beta_cdf()
99 digamma_sum[n] = digamma(alpha_vec.val(n) + beta_vec.val(n)); in beta_cdf()
112 const T_partials_return alpha_dbl = alpha_vec.val(n); in beta_cdf()
|
| /dports/math/stanmath/math-4.2.0/test/unit/math/prim/prob/ |
| H A D | dirichlet_test.cpp | 44 std::vector<Matrix<double, Dynamic, 1>> alpha_vec(3); in TEST() local
45 alpha_vec[0] = alpha1; in TEST()
46 alpha_vec[1] = alpha2; in TEST()
47 alpha_vec[2] = alpha3; in TEST()
54 EXPECT_FLOAT_EQ(result.sum(), dirichlet_log(theta_vec, alpha_vec)); in TEST()
66 EXPECT_FLOAT_EQ(result.sum(), dirichlet_log(theta1, alpha_vec)); in TEST()
|