Class FusedBatchNorm

Inheritance Relationships

class FusedBatchNorm : public mindspore::ops::BaseOperator

FusedBatchNorm defined Enhanced BatchNorm operator prototype.

Public Functions

void Init(const int64_t mode = 0, const float epsilon = 1e-5, const float momentum = 0.1)

Method to init the op's attributes.

Parameters

mode – [in] Define the mode of batchnorm, which is useless.
epsilon – [in] Define a small value added for numerical stability, which is used to avoid that divisor is zero.
momentum – [in] Define the hyper parameter to compute moving average for running_mean and running_var.

void set_mode(const int64_t mode)

Method to set mode attribute, which has been abandoned.

Parameters: mode – [in] Define the mode of batchnorm, which is useless.

void set_epsilon(const float epsilon)

Method to set epsilon attribute.

Parameters: epsilon – [in] Define a small value added for numerical stability, which is used to avoid that divisor is zero.

void set_momentum(const float momentum)

Method to set momentum attribute.

Parameters: momentum – [in] Define the hyper parameter to compute moving average for running_mean and running_var.

int64_t get_mode() const: Method to get mode attribute, which has been abandoned.

float get_epsilon() const

Method to get epsilon attribute.

float get_momentum() const

Method to get momentum attribute.