AlphaDropout

In-place version of AlphaDropout.

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 ↗

Apple MLX

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

Flax NNX

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