Pooling ======= Pooling layer, performing max or average pooling. **Abstract Signature:** ``Pooling(window_shape: Sequence[int], window_stride: Sequence[int], pooling_type: str)`` .. raw:: html

PyTorch

API: torch.nn.MaxPool2d
Strategy: Direct Mapping

Keras

API: keras.layers.MaxPooling2D
Strategy: Direct Mapping

Flax NNX

API: nnx.max_pool
Strategy: Direct Mapping

PaxML / Praxis

API: praxis.layers.Pooling
Strategy: Direct Mapping