class ivy.neural_net_stateful.layers.Dropout(prob, scale=True)[source]

Bases: ivy.neural_net_stateful.module.Module

__init__(prob, scale=True)[source]

Dropout layer. The layer randomly zeroes some of the elements of the input tensor with probability p using samples from a Bernoull distribution.

  • prob (float) – The probability of zeroing out each array element.

  • scale (bool, optional) – Whether to scale the output by 1/(1-prob), default is True.

Supported Frameworks:

empty jax_logo empty tf_logo empty pytorch_logo empty mxnet_logo empty numpy_logo empty