NativeDropoutΒΆ

Native dropout implementation.

Abstract Signature:

NativeDropout(input: Tensor, p: float, train: bool)

PyTorch

API: torch.native_dropout
Strategy: Direct Mapping

JAX (Core)

API: jax.random.bernoulli
Strategy: Plugin (dropout_macro)

Keras

API: keras.random.dropout
Strategy: Direct Mapping