sum.hpp

Go to the documentation of this file.

#ifndef STAN_MATH_FWD_FUN_SUM_HPP
#define STAN_MATH_FWD_FUN_SUM_HPP
 
#include <stan/math/prim/fun/Eigen.hpp>
#include <stan/math/fwd/core.hpp>
#include <stan/math/fwd/meta.hpp>
#include <stan/math/prim/meta.hpp>
#include <stan/math/prim/fun/sum.hpp>
#include <vector>
 
namespace stan {
namespace math {
 
template <typename T, require_fvar_t<T>* = nullptr>
inline auto sum(const std::vector<T>& m) {
  if (m.size() == 0) {
    return T(0.0);
  }
  std::vector<partials_type_t<T>> vals(m.size());
  std::vector<partials_type_t<T>> tans(m.size());
  for (size_t i = 0; i < m.size(); ++i) {
    vals[i] = m[i].val();
    tans[i] = m[i].d();
  }
  return T(sum(vals), sum(tans));
}
 
template <typename T, require_eigen_vt<is_fvar, T>* = nullptr>
inline value_type_t<T> sum(const T& m) {
  if (m.size() == 0) {
    return 0.0;
  }
  const Eigen::Ref<const plain_type_t<T>>& m_ref = m;
  return {sum(m_ref.val()), sum(m_ref.d())};
}
 
}  // namespace math
}  // namespace stan
#endif