hypergeometric_1F0.hpp

Go to the documentation of this file.

#ifndef STAN_MATH_REV_FUN_HYPERGEOMETRIC_1F0_HPP
#define STAN_MATH_REV_FUN_HYPERGEOMETRIC_1F0_HPP
 
#include <stan/math/rev/core.hpp>
#include <stan/math/rev/fun/log1m.hpp>
#include <stan/math/rev/fun/inv.hpp>
#include <stan/math/rev/fun/value_of.hpp>
#include <stan/math/prim/meta.hpp>
#include <stan/math/prim/fun/hypergeometric_1F0.hpp>
 
namespace stan {
namespace math {
 
template <typename Ta, typename Tz,
          require_all_stan_scalar_t<Ta, Tz>* = nullptr,
          require_any_var_t<Ta, Tz>* = nullptr>
inline var hypergeometric_1F0(const Ta& a, const Tz& z) {
  double a_val = value_of(a);
  double z_val = value_of(z);
  double rtn = hypergeometric_1F0(a_val, z_val);
  return make_callback_var(rtn, [rtn, a, z, a_val, z_val](auto& vi) mutable {
    if constexpr (is_autodiff_v<Ta>) {
      a.adj() += vi.adj() * -rtn * log1m(z_val);
    }
    if constexpr (is_autodiff_v<Tz>) {
      z.adj() += vi.adj() * rtn * a_val * inv(1 - z_val);
    }
  });
}
 
}  // namespace math
}  // namespace stan
#endif