Automatic Differentiation
 
Loading...
Searching...
No Matches
square.hpp
Go to the documentation of this file.
1#ifndef STAN_MATH_PRIM_FUN_SQUARE_HPP
2#define STAN_MATH_PRIM_FUN_SQUARE_HPP
3
8#include <cmath>
9
10namespace stan {
11namespace math {
12
26template <typename T, require_arithmetic_t<T>* = nullptr>
27inline double square(const T x) {
28 return std::pow(x, 2);
29}
30
38struct square_fun {
39 template <typename T>
40 static inline auto fun(T&& x) {
41 return square(std::forward<T>(x));
42 }
43};
44
52template <typename Container, require_ad_container_t<Container>* = nullptr>
53inline auto square(Container&& x) {
55 std::forward<Container>(x));
56}
57
66template <typename Container,
68inline auto square(Container&& x) {
69 return apply_vector_unary<Container>::apply(
70 std::forward<Container>(x), [](auto&& v) { return v.array().square(); });
71}
72
73} // namespace math
74} // namespace stan
75
76#endif
require_t< container_type_check_base< is_container, base_type_t, TypeCheck, Check... > > require_container_bt
Require type satisfies is_container.
fvar< T > square(const fvar< T > &x)
Definition square.hpp:12
The lgamma implementation in stan-math is based on either the reentrant safe lgamma_r implementation ...
Base template class for vectorization of unary scalar functions defined by a template class F to a sc...
static auto fun(T &&x)
Definition square.hpp:40
Structure to wrap square() so that it can be vectorized.
Definition square.hpp:38