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.