mxnet
|
#include <initializer.h>
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) |
|
inline |
|
inlineoverridevirtual |
Reimplemented from mxnet::cpp::Initializer.
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 |