Class BatchNorm

Inheritance Relationships

Base Type

  • public mindspore::ops::BaseOperator

Class Documentation

class BatchNorm : public mindspore::ops::BaseOperator

Batch Normalization for input data and updated parameters. Refer to Python API mindspore.ops.BatchNorm for more details.

Public Functions

inline BatchNorm()

Constructor.

void Init(const bool is_training = false, const float epsilon = 1e-5, const float momentun = 0.1, const Format &format = NCHW)

Init. Refer to the parameters of Python API mindspore.ops.BatchNorm for the inputs.

void set_is_training(const bool is_training)

Set is_training.

void set_epsilon(const float epsilon)

Set epsilon.

void set_format(const Format &format)

Set format.

void set_momentum(const float momentum)

Set momentum.

bool get_is_training() const

Get is_training.

Returns

is_training.

float get_epsilon() const

Get epsilon.

Returns

epsilon.

Format get_format() const

Get format.

Returns

format.

float get_momentum() const

Get momentum.

Returns

momentum.