atan.hpp

Go to the documentation of this file.

#ifndef STAN_MATH_REV_FUN_ATAN_HPP
#define STAN_MATH_REV_FUN_ATAN_HPP
 
#include <stan/math/rev/core.hpp>
#include <stan/math/rev/fun/hypot.hpp>
#include <stan/math/rev/meta.hpp>
#include <stan/math/rev/fun/abs.hpp>
#include <stan/math/rev/fun/cos.hpp>
#include <stan/math/rev/fun/atanh.hpp>
#include <stan/math/rev/fun/log.hpp>
#include <stan/math/rev/fun/sqrt.hpp>
#include <stan/math/rev/fun/is_inf.hpp>
#include <stan/math/rev/fun/is_nan.hpp>
#include <stan/math/prim/fun/atan.hpp>
#include <cmath>
#include <complex>
 
namespace stan {
namespace math {
 
inline var atan(const var& x) {
  return make_callback_var(std::atan(x.val()), [x](const auto& vi) mutable {
    x.adj() += vi.adj() / (1.0 + (x.val() * x.val()));
  });
}
 
template <typename VarMat, require_var_matrix_t<VarMat>* = nullptr>
inline auto atan(const VarMat& x) {
  return make_callback_var(
      x.val().array().atan().matrix(), [x](const auto& vi) mutable {
        x.adj().array()
            += vi.adj().array() / (1.0 + (x.val().array().square()));
      });
}
inline std::complex<var> atan(const std::complex<var>& z) {
  return stan::math::internal::complex_atan(z);
}
 
}  // namespace math
}  // namespace stan
#endif