DiSMEC++
dismec::init::NumpyInitializationStrategy Class Reference
Inheritance diagram for dismec::init::NumpyInitializationStrategy:
dismec::init::WeightInitializationStrategy

Public Member Functions

 NumpyInitializationStrategy (std::shared_ptr< const WeightMatrix >, std::shared_ptr< const DenseRealVector > biases)
 
std::unique_ptr< WeightsInitializermake_initializer (const std::shared_ptr< const GenericFeatureMatrix > &features) const override
 Creats a new, thread local WeightsInitializer. More...
 
- Public Member Functions inherited from dismec::init::WeightInitializationStrategy
virtual ~WeightInitializationStrategy ()=default
 

Private Attributes

std::shared_ptr< const WeightMatrix > m_WeightMatrix
 
std::shared_ptr< const DenseRealVectorm_BiasVector
 

Detailed Description

Definition at line 38 of file numpy.cpp.

Constructor & Destructor Documentation

◆ NumpyInitializationStrategy()

NumpyInitializationStrategy::NumpyInitializationStrategy ( std::shared_ptr< const WeightMatrix >  weights,
std::shared_ptr< const DenseRealVector biases 
)
explicit

Definition at line 48 of file numpy.cpp.

Member Function Documentation

◆ make_initializer()

std::unique_ptr< WeightsInitializer > NumpyInitializationStrategy::make_initializer ( const std::shared_ptr< const GenericFeatureMatrix > &  features) const
overridevirtual

Creats a new, thread local WeightsInitializer.

This function will be called from the thread in which the returned WeightsInitializer will be used. It gets passed in a numa-local copy of the feature matrix.

Parameters
featuresRead-only reference to the numa-local feature matrix.
Returns
A new WeightsInitializer.

Implements dismec::init::WeightInitializationStrategy.

Definition at line 54 of file numpy.cpp.

References m_BiasVector, and m_WeightMatrix.

Member Data Documentation

◆ m_BiasVector

std::shared_ptr<const DenseRealVector> dismec::init::NumpyInitializationStrategy::m_BiasVector
private

Definition at line 44 of file numpy.cpp.

Referenced by make_initializer().

◆ m_WeightMatrix

std::shared_ptr<const WeightMatrix> dismec::init::NumpyInitializationStrategy::m_WeightMatrix
private

Definition at line 43 of file numpy.cpp.

Referenced by make_initializer().


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