1#ifndef STAN_MATH_OPENCL_REV_SOFTMAX_HPP
2#define STAN_MATH_OPENCL_REV_SOFTMAX_HPP
23 require_all_kernel_expressions_and_none_scalar_t<T>* =
nullptr>
30 A.adj() += elt_multiply(
31 res.val(), (res.adj() - dot_product(res.adj(), res.val())));
Represents an arithmetic matrix on the OpenCL device.
var_value< plain_type_t< T > > make_callback_var(T &&value, F &&functor)
Creates a new var initialized with a callback_vari with a given value and reverse-pass callback funct...
auto softmax(const ColVec &alpha)
The lgamma implementation in stan-math is based on either the reentrant safe lgamma_r implementation ...