mxnet
Public Types | Public Member Functions | Public Attributes | List of all members
mxnet::cpp::Xavier Class Reference

#include <initializer.h>

Inheritance diagram for mxnet::cpp::Xavier:
Inheritance graph
Collaboration diagram for mxnet::cpp::Xavier:
Collaboration graph

Public Types

enum  RandType { gaussian, uniform }
 
enum  FactorType { avg, in, out }
 

Public Member Functions

 Xavier (RandType rand_type=gaussian, FactorType factor_type=avg, float magnitude=3)
 
void operator() (const std::string &name, NDArray *arr) override
 

Public Attributes

enum mxnet::cpp::Xavier::RandType rand_type
 
enum mxnet::cpp::Xavier::FactorType factor_type
 
float magnitude
 

Additional Inherited Members

- Static Public Member Functions inherited from mxnet::cpp::Initializer
static bool StringStartWith (const std::string &name, const std::string &check_str)
 
static bool StringEndWith (const std::string &name, const std::string &check_str)
 
- Protected Member Functions inherited from mxnet::cpp::Initializer
virtual void InitBilinear (NDArray *arr)
 
virtual void InitZero (NDArray *arr)
 
virtual void InitOne (NDArray *arr)
 
virtual void InitBias (NDArray *arr)
 
virtual void InitGamma (NDArray *arr)
 
virtual void InitBeta (NDArray *arr)
 
virtual void InitWeight (NDArray *arr)
 
virtual void InitDefault (NDArray *arr)
 

Member Enumeration Documentation

Enumerator
avg 
in 
out 
Enumerator
gaussian 
uniform 

Constructor & Destructor Documentation

mxnet::cpp::Xavier::Xavier ( RandType  rand_type = gaussian,
FactorType  factor_type = avg,
float  magnitude = 3 
)
inline

Member Function Documentation

void mxnet::cpp::Xavier::operator() ( const std::string &  name,
NDArray arr 
)
inlineoverridevirtual

Reimplemented from mxnet::cpp::Initializer.

Member Data Documentation

enum mxnet::cpp::Xavier::FactorType mxnet::cpp::Xavier::factor_type
float mxnet::cpp::Xavier::magnitude
enum mxnet::cpp::Xavier::RandType mxnet::cpp::Xavier::rand_type

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