Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions
BatchNormalization Class Reference
Inheritance diagram for BatchNormalization:

Public Member Functions

 BatchNormalization (float epsilon=1e-5f, float momentum=0.9f, bool spatial=true, bool isTest=false)
 
 NOMNIGRAPH_DEFINE_NN_RTTI (BatchNormalization)
 
float getEpsilon () const
 
float getMomentum () const
 
bool getSpatial () const
 
bool getIsTest () const
 
void setEpsilon (float epsilon)
 
void setMomentum (float momentum)
 
void setSpatial (bool spatial)
 
void setIsTest (bool isTest)
 

Detailed Description

Definition at line 480 of file OpClasses.h.


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