Automatic Differentiation
 
Loading...
Searching...
No Matches
abs.hpp
Go to the documentation of this file.
1#ifndef STAN_MATH_PRIM_FUN_ABS_HPP
2#define STAN_MATH_PRIM_FUN_ABS_HPP
3
10#include <cmath>
11#include <complex>
12
13namespace stan {
14namespace math {
15
24template <typename T, require_arithmetic_t<T>* = nullptr>
25inline auto abs(T&& x) {
26 return std::abs(x);
27}
28
37template <typename T, require_complex_bt<std::is_arithmetic, T>* = nullptr>
38inline auto abs(T&& x) {
39 return std::hypot(x.real(), x.imag());
40}
41
50struct abs_fun {
51 template <typename T>
52 static inline auto fun(T&& x) {
53 return abs(std::forward<T>(x));
54 }
55};
56
65template <typename Container, require_ad_container_t<Container>* = nullptr>
66inline auto abs(Container&& x) {
68 std::forward<Container>(x));
69}
70
79template <typename Container,
81inline auto abs(Container&& x) {
82 return apply_vector_unary<Container>::apply(
83 std::forward<Container>(x), [](auto&& v) { return v.array().abs(); });
84}
85
86namespace internal {
94template <typename V>
95inline V complex_abs(const std::complex<V>& z) {
96 return hypot(z.real(), z.imag());
97}
98} // namespace internal
99
100} // namespace math
101} // namespace stan
102
103#endif
require_t< container_type_check_base< is_container, base_type_t, TypeCheck, Check... > > require_container_bt
Require type satisfies is_container.
V complex_abs(const std::complex< V > &z)
Return the absolute value of the complex argument.
Definition abs.hpp:95
fvar< T > hypot(const fvar< T > &x1, const fvar< T > &x2)
Return the length of the hypotenuse of a right triangle with opposite and adjacent side lengths given...
Definition hypot.hpp:26
fvar< T > abs(const fvar< T > &x)
Definition abs.hpp:15
The lgamma implementation in stan-math is based on either the reentrant safe lgamma_r implementation ...
static auto fun(T &&x)
Definition abs.hpp:52
Return elementwise absolute value of the specified real-valued container.
Definition abs.hpp:50
Base template class for vectorization of unary scalar functions defined by a template class F to a sc...