Automatic Differentiation
 
Loading...
Searching...
No Matches
log1m_inv_logit.hpp
Go to the documentation of this file.
1#ifndef STAN_MATH_REV_FUN_LOG1M_INV_LOGIT_HPP
2#define STAN_MATH_REV_FUN_LOG1M_INV_LOGIT_HPP
3
8
9namespace stan {
10namespace math {
11
20template <typename T, require_stan_scalar_or_eigen_t<T>* = nullptr>
21inline auto log1m_inv_logit(const var_value<T>& u) {
22 auto precomp_inv_logit = to_arena(as_array_or_scalar(-inv_logit(u.val())));
23 return make_callback_var(
24 log1m_inv_logit(u.val()), [u, precomp_inv_logit](auto& vi) mutable {
25 as_array_or_scalar(u.adj())
26 += as_array_or_scalar(vi.adj()) * precomp_inv_logit;
27 });
28}
29
30} // namespace math
31} // namespace stan
32#endif
T as_array_or_scalar(T &&v)
Returns specified input value.
var_value< plain_type_t< T > > make_callback_var(T &&value, F &&functor)
Creates a new var initialized with a callback_vari with a given value and reverse-pass callback funct...
arena_t< T > to_arena(const T &a)
Converts given argument into a type that either has any dynamic allocation on AD stack or schedules i...
Definition to_arena.hpp:25
fvar< T > inv_logit(const fvar< T > &x)
Returns the inverse logit function applied to the argument.
Definition inv_logit.hpp:20
fvar< T > log1m_inv_logit(const fvar< T > &x)
Return the natural logarithm of one minus the inverse logit of the specified argument.
The lgamma implementation in stan-math is based on either the reentrant safe lgamma_r implementation ...