ATLAS Offline Software
Loading...
Searching...
No Matches
lwtDev::NormalizationLayer Class Reference

Normalization layer /// https://arxiv.org/abs/1502.03167 ///. More...

#include <Stack.h>

Inheritance diagram for lwtDev::NormalizationLayer:
Collaboration diagram for lwtDev::NormalizationLayer:

Public Member Functions

 NormalizationLayer (const VectorXd &W, const VectorXd &b)
virtual VectorXd compute (const VectorXd &) const override

Private Attributes

VectorXd m_W
VectorXd m_b

Detailed Description

Normalization layer /// https://arxiv.org/abs/1502.03167 ///.

Definition at line 139 of file Stack.h.

Constructor & Destructor Documentation

◆ NormalizationLayer()

lwtDev::NormalizationLayer::NormalizationLayer ( const VectorXd & W,
const VectorXd & b )

Definition at line 241 of file Stack.cxx.

242 :
243 m_W(W), m_b(b)
244 {
245 }

Member Function Documentation

◆ compute()

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

Implements lwtDev::ILayer.

Definition at line 246 of file Stack.cxx.

246 {
247 VectorXd shift = in + m_b ;
248 return m_W.cwiseProduct(shift);
249 }

Member Data Documentation

◆ m_b

VectorXd lwtDev::NormalizationLayer::m_b
private

Definition at line 148 of file Stack.h.

◆ m_W

VectorXd lwtDev::NormalizationLayer::m_W
private

Definition at line 147 of file Stack.h.


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