Stan Math Library
4.9.0
Automatic Differentiation
|
#include <stan/math/opencl/prim/log_softmax.hpp>
#include <stan/math/opencl/kernel_generator.hpp>
#include <stan/math/rev/core.hpp>
#include <stan/math/rev/fun/value_of.hpp>
Go to the source code of this file.
Namespaces | |
namespace | stan |
The lgamma implementation in stan-math is based on either the reentrant safe lgamma_r implementation from C or the boost::math::lgamma implementation. | |
namespace | stan::math |
Matrices and templated mathematical functions. | |
Functions | |
template<typename T , require_all_kernel_expressions_and_none_scalar_t< T > * = nullptr> | |
var_value< matrix_cl< double > > | stan::math::log_softmax (const var_value< T > &A) |
Return log of the softmax of the specified vector. | |