libdl  0.0.1
Simple yet powerful deep learning
Loading...
Searching...
No Matches
dl::LayerNorm Class Referencefinal

Implements layer normalization as proposed by [1]. More...

#include <layernorm.hpp>

Inheritance diagram for dl::LayerNorm:
Collaboration diagram for dl::LayerNorm:

Public Member Functions

 LayerNorm (Shape normShape, const Device &device=Device::getDefault()) noexcept
 
virtual TensorPtr forward (TensorPtr input) noexcept override
 

Detailed Description

Implements layer normalization as proposed by [1].

Definition at line 12 of file layernorm.hpp.


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