mean.hpp

Go to the documentation of this file.

#ifndef STAN__MATH__MATRIX__MEAN_HPP
#define STAN__MATH__MATRIX__MEAN_HPP

#include <vector>
#include <boost/math/tools/promotion.hpp>
#include <stan/math/matrix/Eigen.hpp>
#include <stan/math/error_handling/matrix/check_nonzero_size.hpp>

namespace stan {
  namespace math {
    
    template <typename T>
    inline 
    typename boost::math::tools::promote_args<T>::type
    mean(const std::vector<T>& v) {
      stan::math::check_nonzero_size("mean(%1%)",v,"v", (double*)0);
      T sum(v[0]);
      for (size_t i = 1; i < v.size(); ++i)
        sum += v[i];
      return sum / v.size();
    }

    template <typename T, int R, int C>
    inline  
    typename boost::math::tools::promote_args<T>::type
    mean(const Eigen::Matrix<T,R,C>& m) {
      stan::math::check_nonzero_size("mean(%1%)",m,"m", (double*)0);
      return m.mean();
    }    
    
  }
}
#endif