ATLAS Offline Software
Public Member Functions | Private Attributes | List of all members
lwtDev::UnaryActivationLayer Class Reference

#include <Stack.h>

Inheritance diagram for lwtDev::UnaryActivationLayer:
Collaboration diagram for lwtDev::UnaryActivationLayer:

Public Member Functions

 UnaryActivationLayer (ActivationConfig)
 
virtual VectorXd compute (const VectorXd &) const override
 

Private Attributes

std::function< double(double)> m_func
 

Detailed Description

Definition at line 91 of file Stack.h.

Constructor & Destructor Documentation

◆ UnaryActivationLayer()

lwtDev::UnaryActivationLayer::UnaryActivationLayer ( ActivationConfig  act)

Definition at line 174 of file Stack.cxx.

174  :
175  m_func(get_activation(act))
176  {
177  }

Member Function Documentation

◆ compute()

VectorXd lwtDev::UnaryActivationLayer::compute ( const VectorXd &  in) const
overridevirtual

Implements lwtDev::ILayer.

Definition at line 178 of file Stack.cxx.

178  {
179  return in.unaryExpr(m_func);
180  }

Member Data Documentation

◆ m_func

std::function<double(double)> lwtDev::UnaryActivationLayer::m_func
private

Definition at line 97 of file Stack.h.


The documentation for this class was generated from the following files:
lwtDev::UnaryActivationLayer::m_func
std::function< double(double)> m_func
Definition: Stack.h:97
lwtDev::get_activation
std::function< double(double)> get_activation(lwtDev::ActivationConfig act)
Definition: Stack.cxx:671