AlphaDropout

Applies Alpha Dropout to the input.

Abstract Signature:

AlphaDropout(p: float = 0.5)

PyTorch

API: torch.nn.AlphaDropout
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: flax.linen.AlphaDropout
Strategy: Direct Mapping
Official Docs ↗

Keras

API: keras.layers.AlphaDropout
Strategy: Direct Mapping
Official Docs ↗

TensorFlow

API: keras.layers.AlphaDropout
Strategy: Direct Mapping
Official Docs ↗

Apple MLX

API: mx.nn.AlphaDropout
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: flax.linen.AlphaDropout
Strategy: Direct Mapping
Official Docs ↗

PaxML / Praxis

API: praxis.layers.AlphaDropout
Strategy: Macro 'praxis.layers.AlphaDropout(keep_prob=1.0-{p})'