Dropout

In-place dropout.

Abstract Signature:

Dropout_(input: Tensor, p: float = 0.5, training: bool = True)

PyTorchJAX (Core)KerasTensorFlowFlax NNX

PyTorch

API: torch.nn.functional.dropout_

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: —

Strategy: Plugin (inject_prng)

Keras

API: keras.random.dropout

Strategy: Direct Mapping

Official Docs ↗

TensorFlow

API: tf.nn.dropout

Strategy: Direct Mapping

Official Docs ↗

Flax NNX

API: —

Strategy: Plugin (inject_prng)