lbann::normal_initializer< TensorDataType > Class Template Reference

Draw weights values from a normal random distribution. More...

#include <initializer.hpp>

Inheritance diagram for lbann::normal_initializer< TensorDataType >:

Collaboration diagram for lbann::normal_initializer< TensorDataType >:

Public Types
Public Types
using	AbsDistMatrixType = El::AbstractDistMatrix< TensorDataType >
	The tensor type expected in this object. More...

Public Member Functions
	normal_initializer (TensorDataType mean=El::TypeTraits< TensorDataType >::Zero(), TensorDataType standard_deviation=El::TypeTraits< TensorDataType >::One())
std::string	get_type () const override
description	get_description () const override
void	fill (AbsDistMatrixType &matrix) override
void	write_proto (lbann_data::Initializer &init) const final
	Add initializer data to prototext. More...
Public Member Functions inherited from lbann::Cloneable< normal_initializer< TensorDataType >, data_type_weights_initializer< TensorDataType > >
std::unique_ptr< normal_initializer< TensorDataType > >	clone () const
	Return an exception-safe, memory-safe copy of this object. More...

Private Attributes
TensorDataType	m_mean
TensorDataType	m_standard_deviation

Detailed Description

template<typename TensorDataType>
class lbann::normal_initializer< TensorDataType >

Draw weights values from a normal random distribution.

Definition at line 221 of file initializer.hpp.

Member Typedef Documentation

AbsDistMatrixType

template<typename TensorDataType >

using lbann::normal_initializer< TensorDataType >::AbsDistMatrixType = El::AbstractDistMatrix<TensorDataType>

The tensor type expected in this object.

Definition at line 230 of file initializer.hpp.

Constructor & Destructor Documentation

normal_initializer()

template<typename TensorDataType >

lbann::normal_initializer< TensorDataType >::normal_initializer ( TensorDataType  mean = El::TypeTraits<TensorDataType>::Zero(), TensorDataType  standard_deviation = El::TypeTraits<TensorDataType>::One()  )

inline

Definition at line 235 of file initializer.hpp.

Member Function Documentation

fill()

template<typename TensorDataType >

void lbann::normal_initializer< TensorDataType >::fill ( AbsDistMatrixType &  matrix )

override

get_description()

template<typename TensorDataType >

description lbann::normal_initializer< TensorDataType >::get_description ( ) const

override

get_type()

template<typename TensorDataType >

std::string lbann::normal_initializer< TensorDataType >::get_type ( ) const

inlineoverride

Definition at line 240 of file initializer.hpp.

Here is the call graph for this function:

write_proto()

template<typename TensorDataType >

void lbann::normal_initializer< TensorDataType >::write_proto ( lbann_data::Initializer &  init ) const

final

Add initializer data to prototext.

Member Data Documentation

m_mean

template<typename TensorDataType >

TensorDataType lbann::normal_initializer< TensorDataType >::m_mean

private

Normal distribution mean.

Definition at line 249 of file initializer.hpp.

m_standard_deviation

template<typename TensorDataType >

TensorDataType lbann::normal_initializer< TensorDataType >::m_standard_deviation

private

Normal distribution standard deviation.

Definition at line 251 of file initializer.hpp.

---

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

• initializer.hpp