mlpack
Public Member Functions | List of all members
mlpack::ann::NetworkInitialization< InitializationRuleType, CustomLayers > Class Template Reference

This class is used to initialize the network with the given initialization rule. More...

#include <network_init.hpp>

Public Member Functions

 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 > &parameter, size_t parameterOffset=0)
 Initialize the specified network and store the results in the given parameter. More...
 

Detailed Description

template<typename InitializationRuleType, typename... CustomLayers>
class mlpack::ann::NetworkInitialization< InitializationRuleType, CustomLayers >

This class is used to initialize the network with the given initialization rule.

Constructor & Destructor Documentation

◆ NetworkInitialization()

template<typename InitializationRuleType, typename... CustomLayers>
mlpack::ann::NetworkInitialization< InitializationRuleType, CustomLayers >::NetworkInitialization ( const InitializationRuleType &  initializeRule = InitializationRuleType())
inline

Use the given initialization rule to initialize the specified network.

Parameters
initializeRuleRule to initialize the given network.

Member Function Documentation

◆ 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
networkNetwork that should be initialized.
parameterThe network parameter.
parameterOffsetOffset for network paramater, default 0.

The documentation for this class was generated from the following file: