prob_constrain.hpp

Go to the documentation of this file.

#ifndef STAN_MATH_PRIM_CONSTRAINT_PROB_CONSTRAIN_HPP
#define STAN_MATH_PRIM_CONSTRAINT_PROB_CONSTRAIN_HPP
 
#include <stan/math/prim/meta.hpp>
#include <stan/math/prim/fun/log_inv_logit.hpp>
#include <stan/math/prim/fun/exp.hpp>
#include <stan/math/prim/fun/log1m_inv_logit.hpp>
#include <cmath>
 
namespace stan {
namespace math {
 
template <typename T>
inline auto prob_constrain(T&& x) {
  return inv_logit(std::forward<T>(x));
}
 
template <typename T, typename Lp>
inline auto prob_constrain(T&& x, Lp& lp) {
  plain_type_t<T> log_inv_logit_x = log_inv_logit(x);
  lp += log_inv_logit_x + log1m_inv_logit(std::forward<T>(x));
  return exp(std::move(log_inv_logit_x));
}
 
template <bool Jacobian, typename T, typename Lp>
inline auto prob_constrain(T&& x, Lp& lp) {
  if (Jacobian) {
    return prob_constrain(std::forward<T>(x), lp);
  } else {
    return prob_constrain(std::forward<T>(x));
  }
}
}  // namespace math
}  // namespace stan
 
#endif