mindspore.ops.Dropout
- class mindspore.ops.Dropout(keep_prob=0.5, Seed0=0, Seed1=0)[source]
During training, randomly zeroes some of the elements of the input tensor with probability 1-keep_prob from a Bernoulli distribution.
- Parameters
- Inputs:
x (Tensor) - The input of Dropout, a Tensor of any shape with data type of float16 or float32.
- Outputs:
output (Tensor) - With the same shape and data type as x.
mask (Tensor) - With the same shape as x.
- Raises
- Supported Platforms:
Ascend
GPU
CPU
Examples
>>> dropout = ops.Dropout(keep_prob=0.5) >>> x = Tensor(((20, 16), (50, 50)), mindspore.float32) >>> output, mask = dropout(x) >>> print(output.shape) (2, 2)