math/opencl_2prim_2mean_8hpp_source.html

#ifndef STAN_MATH_OPENCL_PRIM_MEAN_HPP

#define STAN_MATH_OPENCL_PRIM_MEAN_HPP

#ifdef STAN_OPENCL


#include <stan/math/opencl/kernel_generator.hpp>

#include <stan/math/opencl/prim/sum.hpp>


namespace stan {

namespace math {


template <typename T,

          require_nonscalar_prim_or_rev_kernel_expression_t<T>* = nullptr>

scalar_type_t<T> mean(const T& m) {

  check_nonzero_size("mean", "m", m);

  return sum(m) / m.size();

}


}  // namespace math

}  // namespace stan


#endif

#endif

kernel_generator.hpp

stan::math::mean
scalar_type_t< T > mean(const T &m)
Returns the sample mean (i.e., average) of the coefficients in the specified std vector,...
Definition mean.hpp:20

stan::math::sum
fvar< T > sum(const std::vector< fvar< T > > &m)
Return the sum of the entries of the specified standard vector.
Definition sum.hpp:22

stan::math::check_nonzero_size
void check_nonzero_size(const char *function, const char *name, const T_y &y)
Check if the specified matrix/vector is of non-zero size.
Definition check_nonzero_size.hpp:22

stan::scalar_type_t
typename scalar_type< T >::type scalar_type_t
Definition scalar_type.hpp:25

stan
The lgamma implementation in stan-math is based on either the reentrant safe lgamma_r implementation ...
Definition fvar.hpp:9

sum.hpp