hypergeometric_2F1.hpp

Go to the documentation of this file.

#ifndef STAN_MATH_REV_FUN_HYPERGEOMETRIC_2F1_HPP
#define STAN_MATH_REV_FUN_HYPERGEOMETRIC_2F1_HPP
 
#include <stan/math/prim/meta.hpp>
#include <stan/math/rev/core.hpp>
#include <stan/math/prim/fun/hypergeometric_2F1.hpp>
#include <stan/math/prim/fun/grad_2F1.hpp>
 
namespace stan {
namespace math {
 
template <typename Ta1, typename Ta2, typename Tb, typename Tz,
          require_all_stan_scalar_t<Ta1, Ta2, Tb, Tz>* = nullptr,
          require_any_var_t<Ta1, Ta2, Tb, Tz>* = nullptr>
inline return_type_t<Ta1, Ta1, Tb, Tz> hypergeometric_2F1(const Ta1& a1,
                                                          const Ta2& a2,
                                                          const Tb& b,
                                                          const Tz& z) {
  double a1_dbl = value_of(a1);
  double a2_dbl = value_of(a2);
  double b_dbl = value_of(b);
  double z_dbl = value_of(z);
 
  return make_callback_var(
      hypergeometric_2F1(a1_dbl, a2_dbl, b_dbl, z_dbl),
      [a1, a2, b, z](auto& vi) mutable {
        auto grad_tuple = grad_2F1(a1, a2, b, z);
 
        if (!is_constant<Ta1>::value) {
          forward_as<var>(a1).adj() += vi.adj() * std::get<0>(grad_tuple);
        }
        if (!is_constant<Ta2>::value) {
          forward_as<var>(a2).adj() += vi.adj() * std::get<1>(grad_tuple);
        }
        if (!is_constant<Tb>::value) {
          forward_as<var>(b).adj() += vi.adj() * std::get<2>(grad_tuple);
        }
        if (!is_constant<Tz>::value) {
          forward_as<var>(z).adj() += vi.adj() * std::get<3>(grad_tuple);
        }
      });
}
}  // namespace math
}  // namespace stan
#endif