dropout

tfsnippet.layers.dropout(*args, **kwargs)

Apply dropout on input.

Parameters:

• input (Tensor) – The input tensor.
• rate (float or tf.Tensor) – The rate of dropout.
• noise_shape (tuple[int] or tf.Tensor) – Shape of the noise. If not specified, use the shape of input.
• training (bool or tf.Tensor) – Whether or not the model is under training stage?
• name (str) – Default name of the name scope. If not specified, generate one according to the method name.

Returns:

The dropout transformed tensor.

Return type:

tf.Tensor