Dropout¶
-
class
Dropout(rate, axes=(), **kwargs)¶ Applies Dropout to the input.
Dropout consists in randomly setting a fraction rate of input units to 0 at each update during training time, which helps prevent overfitting.
Parameters: - rate : float
Fraction of the input units to drop. Must be a number between 0 and 1.
- axes : tuple of int, default ()
The axes on which dropout mask is shared. If empty, regular dropout is applied.
- Inputs:
- data: input tensor with arbitrary shape.
- Outputs:
- out: output tensor with the same shape as data.