Stan Math Library
2.15.0
reverse mode automatic differentiation
stan
math
rev
mat
fun
grad.hpp
Go to the documentation of this file.
1
#ifndef STAN_MATH_REV_MAT_FUN_GRAD_HPP
2
#define STAN_MATH_REV_MAT_FUN_GRAD_HPP
3
4
#include <
stan/math/prim/mat/fun/Eigen.hpp
>
5
#include <
stan/math/rev/mat/fun/Eigen_NumTraits.hpp
>
6
#include <
stan/math/rev/core.hpp
>
7
8
namespace
stan
{
9
namespace
math {
10
24
inline
void
grad
(
var
& v,
25
Eigen::Matrix<var, Eigen::Dynamic, 1>& x,
26
Eigen::VectorXd& g) {
27
grad
(v.
vi_
);
28
g.resize(x.size());
29
for
(
int
i = 0; i < x.size(); ++i)
30
g(i) = x(i).vi_->adj_;
31
}
32
33
}
34
}
35
#endif
core.hpp
Eigen_NumTraits.hpp
stan
Definition:
log_sum_exp.hpp:8
stan::math::var
Independent (input) and dependent (output) variables for gradients.
Definition:
var.hpp:30
stan::math::grad
static void grad(vari *vi)
Compute the gradient for all variables starting from the specified root variable implementation.
Definition:
grad.hpp:30
stan::math::var::vi_
vari * vi_
Pointer to the implementation of this variable.
Definition:
var.hpp:42
Eigen.hpp
[
Stan Home Page
]
© 2011–2016, Stan Development Team.