Stan Math Library  2.20.0
reverse mode automatic differentiation
Public Member Functions | List of all members
stan::math::internal::softmax_op Class Reference

#include <softmax.hpp>

Public Member Functions

 softmax_op ()
 
template<std::size_t size>
Eigen::VectorXd operator() (const std::array< bool, size > &needs_adj, const Eigen::VectorXd &alpha)
 
template<std::size_t size>
std::tuple< Eigen::VectorXd > multiply_adjoint_jacobian (const std::array< bool, size > &needs_adj, const Eigen::VectorXd &adj) const
 

Detailed Description

Definition at line 15 of file softmax.hpp.

Constructor & Destructor Documentation

◆ softmax_op()

stan::math::internal::softmax_op::softmax_op ( )
inline

Definition at line 20 of file softmax.hpp.

Member Function Documentation

◆ multiply_adjoint_jacobian()

template<std::size_t size>
std::tuple<Eigen::VectorXd> stan::math::internal::softmax_op::multiply_adjoint_jacobian ( const std::array< bool, size > &  needs_adj,
const Eigen::VectorXd &  adj 
) const
inline

Definition at line 50 of file softmax.hpp.

◆ operator()()

template<std::size_t size>
Eigen::VectorXd stan::math::internal::softmax_op::operator() ( const std::array< bool, size > &  needs_adj,
const Eigen::VectorXd &  alpha 
)
inline

Definition at line 29 of file softmax.hpp.


The documentation for this class was generated from the following file:

     [ Stan Home Page ] © 2011–2018, Stan Development Team.