Pooling

Pooling layer, performing max or average pooling.

Abstract Signature:

Pooling(window_shape: Sequence[int], window_stride: Sequence[int], pooling_type: str)

PyTorch

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

Keras

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

Flax NNX

API: nnx.max_pool
Strategy: Direct Mapping
Official Docs ↗

PaxML / Praxis

API: praxis.layers.Pooling
Strategy: Direct Mapping
Official Docs ↗