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().