Class Dropout
Defined in File dropout.h
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.
- Returns
keep_prob.
-
inline Dropout()