math/prim_2fun_2sub__col_8hpp_source.html

#ifndef STAN_MATH_PRIM_FUN_SUB_COL_HPP

#define STAN_MATH_PRIM_FUN_SUB_COL_HPP


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

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


namespace stan {

namespace math {


template <

    typename T, require_matrix_t<T>* = nullptr,

    require_all_not_nonscalar_prim_or_rev_kernel_expression_t<T>* = nullptr>

inline auto sub_col(const T& m, size_t i, size_t j, size_t nrows) {

  check_row_index("sub_col", "i", m, i);

  if (nrows > 0) {

    check_row_index("sub_col", "i+nrows-1", m, i + nrows - 1);

  }

  check_column_index("sub_col", "j", m, j);

  return m.col(j - 1).segment(i - 1, nrows);

}


}  // namespace math

}  // namespace stan


#endif

Eigen.hpp

stan::math::check_row_index
void check_row_index(const char *function, const char *name, const T_y &y, size_t i)
Check if the specified index is a valid row of the matrix This check is 1-indexed by default.
Definition check_row_index.hpp:25

stan::math::check_column_index
void check_column_index(const char *function, const char *name, const T_y &y, size_t i)
Check if the specified index is a valid column of the matrix.
Definition check_column_index.hpp:30

stan::math::sub_col
auto sub_col(T_x &&x, size_t i, size_t j, size_t nrows)
Return a 1 x ncols subrow starting at (i-1, j-1).
Definition sub_col.hpp:23

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

err.hpp