This class is used to initialize the network with the given initialization rule.
More...
#include <network_init.hpp>
|
| NetworkInitialization (const InitializationRuleType &initializeRule=InitializationRuleType()) |
| Use the given initialization rule to initialize the specified network. More...
|
|
template<typename eT > |
void | Initialize (const std::vector< LayerTypes< CustomLayers... > > &network, arma::Mat< eT > ¶meter, size_t parameterOffset=0) |
| Initialize the specified network and store the results in the given parameter. More...
|
|
template<typename InitializationRuleType, typename... CustomLayers>
class mlpack::ann::NetworkInitialization< InitializationRuleType, CustomLayers >
This class is used to initialize the network with the given initialization rule.
◆ NetworkInitialization()
template<typename InitializationRuleType, typename... CustomLayers>
Use the given initialization rule to initialize the specified network.
- Parameters
-
initializeRule | Rule to initialize the given network. |
◆ Initialize()
template<typename InitializationRuleType, typename... CustomLayers>
template<typename eT >
void mlpack::ann::NetworkInitialization< InitializationRuleType, CustomLayers >::Initialize |
( |
const std::vector< LayerTypes< CustomLayers... > > & |
network, |
|
|
arma::Mat< eT > & |
parameter, |
|
|
size_t |
parameterOffset = 0 |
|
) |
| |
|
inline |
Initialize the specified network and store the results in the given parameter.
- Parameters
-
network | Network that should be initialized. |
parameter | The network parameter. |
parameterOffset | Offset for network paramater, default 0. |
The documentation for this class was generated from the following file: