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)
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: