Class Dropout

Inheritance Relationships

Base Type

  • public mindspore::ops::BaseOperator

Class Documentation

class Dropout : public mindspore::ops::BaseOperator

During training, randomly zeroes some of the elements of the input tensor with probability 1-keep_prob / from a Bernoulli distribution. Refer to Python API mindspore.ops.Dropout for more details.

Public Functions

inline Dropout()

Constructor.

void Init(const float keep_prob = 0.5)

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

void set_keep_prob(const float keep_prob)

Set keep_prob.

float get_keep_prob() const

Get keep_prob.

返回

keep_prob.