This class is used to initialize weight matrix with the He initialization rule given by He et. More...
Public Member Functions  
HeInitialization ()  
Initialize the HeInitialization object. More...  
template  
void  Initialize (arma::Mat< eT > &W, const size_t rows, const size_t cols) 
Initialize the elements of the weight matrix with the He initialization rule. More...  
template  
void  Initialize (arma::Mat< eT > &W) 
Initialize the elements of the weight matrix with the He initialization rule. More...  
template  
void  Initialize (arma::Cube< eT > &W, const size_t rows, const size_t cols, const size_t slices) 
Initialize the elements of the specified weight 3rd order tensor with He initialization rule. More...  
template  
void  Initialize (arma::Cube< eT > &W) 
Initialize the elements of the specified weight 3rd order tensor with He initialization rule. More...  
This class is used to initialize weight matrix with the He initialization rule given by He et.
al. for neural networks. The He initialization initializes weights of the neural network to better suit the rectified activation units.
For more information, the following paper can be referred to:
Definition at line 45 of file he_init.hpp.

inline 
Initialize the HeInitialization object.
Definition at line 51 of file he_init.hpp.

inline 
Initialize the elements of the weight matrix with the He initialization rule.
W  Weight matrix to initialize. 
rows  Number of rows. 
cols  Number of columns. 
Definition at line 65 of file he_init.hpp.
Referenced by HeInitialization::Initialize().

inline 
Initialize the elements of the weight matrix with the He initialization rule.
W  Weight matrix to initialize. 
Definition at line 87 of file he_init.hpp.
References Log::Fatal, and if().

inline 
Initialize the elements of the specified weight 3rd order tensor with He initialization rule.
W  Weight matrix to initialize. 
rows  Number of rows. 
cols  Number of columns. 
slices  Number of slices. 
Definition at line 112 of file he_init.hpp.
References HeInitialization::Initialize().

inline 
Initialize the elements of the specified weight 3rd order tensor with He initialization rule.
W  Weight matrix to initialize. 
Definition at line 131 of file he_init.hpp.
References Log::Fatal, and HeInitialization::Initialize().