lwtDev::SoftmaxLayer Class Reference

#include <Stack.h>

Inheritance diagram for lwtDev::SoftmaxLayer:

Collaboration diagram for lwtDev::SoftmaxLayer:

Public Member Functions
virtual VectorXd	compute (const VectorXd &) const override

Detailed Description

Definition at line 100 of file Stack.h.

Member Function Documentation

compute()

VectorXd lwtDev::SoftmaxLayer::compute ( const VectorXd & in ) const

overridevirtual

Implements lwtDev::ILayer.

Definition at line 182 of file Stack.cxx.

                                                         {
    // More numerically stable softmax, as suggested in
    // http://stackoverflow.com/a/34969389
    size_t n_elements = in.rows();
    VectorXd exp(n_elements);
    double max = in.maxCoeff();
    for (size_t iii = 0; iii < n_elements; iii++) {
      exp(iii) = std::exp(in(iii) - max);
    }
    double sum_exp = exp.sum();
    return exp / sum_exp;
  }

---

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

• Stack.h
• Stack.cxx