Dropout2d

Applies 2D Dropout (channel-wise) to the input.

Abstract Signature:

Dropout2d(p: float = 0.5)

PyTorch

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

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.nnx.Dropout
Strategy: Direct Mapping
Official Docs ↗