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...  
void  Initialize (arma::mat &W, const size_t rows, const size_t cols) 
Initialize the elements of the weight matrix with the He initialization rule. More...  
void  Initialize (arma::cube &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...  
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 64 of file he_init.hpp.
Referenced by HeInitialization::Initialize().

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. 
slice  Numbers of slices. 
Definition at line 90 of file he_init.hpp.
References HeInitialization::Initialize().