ATLAS Offline Software
Loading...
Searching...
No Matches
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 :
176 {
177 }
std::function< double(double)> m_func
Definition Stack.h:97
std::function< double(double)> get_activation(lwtDev::ActivationConfig act)
Definition Stack.cxx:671

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: