This class is used to initialize weight matrix with the Lecun Normalization initialization rule. More...
Public Member Functions | |
LecunNormalInitialization () | |
Initialize the LecunNormalInitialization object. More... | |
template < typename eT > | |
void | Initialize (arma::Mat< eT > &W, const size_t rows, const size_t cols) |
Initialize the elements of the weight matrix with the Lecun Normal initialization rule. More... | |
template < typename eT > | |
void | Initialize (arma::Mat< eT > &W) |
Initialize the elements of the weight matrix with the Lecun Normal initialization rule. More... | |
template < typename eT > | |
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 Lecun Normal initialization rule. More... | |
template < typename eT > | |
void | Initialize (arma::Cube< eT > &W) |
Initialize the elements of the specified weight 3rd order tensor with Lecun Normal initialization rule. More... | |
This class is used to initialize weight matrix with the Lecun Normalization initialization rule.
For more information, the following papers can be referred to:
Definition at line 49 of file lecun_normal_init.hpp.
|
inline |
Initialize the LecunNormalInitialization object.
Definition at line 55 of file lecun_normal_init.hpp.
|
inline |
Initialize the elements of the weight matrix with the Lecun Normal initialization rule.
W | Weight matrix to initialize. |
rows | Number of rows. |
cols | Number of columns. |
Definition at line 69 of file lecun_normal_init.hpp.
Referenced by LecunNormalInitialization::Initialize().
|
inline |
Initialize the elements of the weight matrix with the Lecun Normal initialization rule.
W | Weight matrix to initialize. |
Definition at line 93 of file lecun_normal_init.hpp.
References Log::Fatal, and if().
|
inline |
Initialize the elements of the specified weight 3rd order tensor with Lecun Normal initialization rule.
W | Weight matrix to initialize. |
rows | Number of rows. |
cols | Number of columns. |
slices | Number of slices. |
Definition at line 118 of file lecun_normal_init.hpp.
References LecunNormalInitialization::Initialize().
|
inline |
Initialize the elements of the specified weight 3rd order tensor with Lecun Normal initialization rule.
W | Weight matrix to initialize. |
Definition at line 137 of file lecun_normal_init.hpp.
References Log::Fatal, and LecunNormalInitialization::Initialize().