1#ifndef STAN_MATH_PRIM_FUN_LOG_SOFTMAX_HPP
2#define STAN_MATH_PRIM_FUN_LOG_SOFTMAX_HPP
42template <
typename Container, require_st_arithmetic<Container>* =
nullptr,
43 require_container_t<Container>* =
nullptr>
49 std::forward<
decltype(a)>(a),
50 [](
auto&& v) {
return v.array() -
log_sum_exp(v); });
52 to_ref(std::forward<Container>(x)));
auto make_holder(F &&func, Args &&... args)
Calls given function with given arguments.
auto log_softmax(T &&x)
Return the log softmax of the specified vector or container of vectors.
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.
ref_type_t< T && > to_ref(T &&a)
This evaluates expensive Eigen expressions.
constexpr decltype(auto) apply(F &&f, Tuple &&t, PreArgs &&... pre_args)
fvar< T > log_sum_exp(const fvar< T > &x1, const fvar< T > &x2)
The lgamma implementation in stan-math is based on either the reentrant safe lgamma_r implementation ...