DropoutCell¶
-
class
DropoutCell(rate, axes=(), prefix=None, params=None)¶ Applies dropout on input.
Parameters: - rate : float
Percentage of elements to drop out, which is 1 - percentage to retain.
- axes : tuple of int, default ()
The axes on which dropout mask is shared. If empty, regular dropout is applied.
- Inputs:
- data: input tensor with shape (batch_size, size).
- states: a list of recurrent state tensors.
- Outputs:
- out: output tensor with shape (batch_size, size).
- next_states: returns input states directly.