math/one__hot__vector_8hpp_source.html

#ifndef STAN_MATH_PRIM_FUN_ONE_HOT_VECTOR_HPP

#define STAN_MATH_PRIM_FUN_ONE_HOT_VECTOR_HPP


#include <stan/math/prim/err.hpp>

#include <stan/math/prim/fun/Eigen.hpp>


namespace stan {

namespace math {


inline Eigen::VectorXd one_hot_vector(int K, int k) {

  static constexpr const char* function = "one_hot_vector";

  check_positive(function, "size", K);

  check_bounded(function, "k", k, 1, K);


  Eigen::VectorXd ret = Eigen::VectorXd::Zero(K);

  ret(k - 1) = 1;

  return ret;

}


}  // namespace math

}  // namespace stan


#endif

Eigen.hpp

stan::math::check_bounded
void check_bounded(const char *function, const char *name, const T_y &y, const T_low &low, const T_high &high)
Check if the value is between the low and high values, inclusively.
Definition check_bounded.hpp:75

stan::math::check_positive
void check_positive(const char *function, const char *name, const T_y &y)
Check if y is positive.
Definition check_positive.hpp:27

stan::math::one_hot_vector
Eigen::VectorXd one_hot_vector(int K, int k)
Return a vector with 1 in the k-th position and zero elsewhere.
Definition one_hot_vector.hpp:20

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

err.hpp