dropout

ivy.neural_net_functional.layers.dropout(x, prob, scale=True)[source]

Randomly zeroes some of the elements of the input tensor with probability p using samples from a Bernoull distribution.

Parameters
  • x (array) – The input array x to perform dropout on.

  • 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.

Returns

Result array of the linear transformation. [N,∗,out_features]


Supported Frameworks:

empty jax_logo empty tf_logo empty pytorch_logo empty mxnet_logo empty numpy_logo empty