FeatureDropout

Randomly masks out entire channels (spatial dropout).

Abstract Signature:

FeatureDropout(input: Tensor, p: float = 0.5, training: bool = False, inplace: bool = False)

PyTorch

API: torch.feature_dropout
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: —
Strategy: Plugin (inject_prng)

Keras

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

Apple MLX

API: mlx.nn.Dropout2d
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

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