mlpack
master
|
This class is used to initialize the weight matrix with the orthogonal matrix initialization. More...
Public Member Functions | |
OrthogonalInitialization (const double gain=1.0) | |
Initialize the orthogonal matrix initialization rule with the given gain. More... | |
template<typename eT > | |
void | Initialize (arma::Mat< eT > &W, const size_t rows, const size_t cols) |
Initialize the elements of the specified weight matrix with the orthogonal matrix initialization method. 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 the orthogonal matrix initialization method. More... | |
Private Attributes | |
const double | gain |
The number used as gain. More... | |
This class is used to initialize the weight matrix with the orthogonal matrix initialization.
Definition at line 24 of file orthogonal_init.hpp.
|
inline |
Initialize the orthogonal matrix initialization rule with the given gain.
gain | The gain value. |
Definition at line 32 of file orthogonal_init.hpp.
|
inline |
Initialize the elements of the specified weight matrix with the orthogonal matrix initialization method.
W | Weight matrix to initialize. |
rows | Number of rows. |
cols | Number of columns. |
Definition at line 43 of file orthogonal_init.hpp.
References gain.
Referenced by Initialize().
|
inline |
Initialize the elements of the specified weight 3rd order tensor with the orthogonal matrix initialization method.
W | Weight matrix to initialize. |
rows | Number of rows. |
cols | Number of columns. |
slices | Number of slices. |
Definition at line 62 of file orthogonal_init.hpp.
References Initialize().
|
private |
The number used as gain.
Definition at line 75 of file orthogonal_init.hpp.
Referenced by Initialize().