MaxPool2dΒΆ

Applies a 2D max pooling over an input signal.

Abstract Signature:

MaxPool2d(kernel_size: Union[int, Tuple[int, int]], stride: Union[int, Tuple[int, int]])

PyTorch

API: torch.nn.MaxPool2d
Strategy: Direct Mapping

Keras

API: keras.layers.MaxPooling2D
Strategy: Direct Mapping

TensorFlow

API: tf.keras.layers.MaxPooling2D
Strategy: Direct Mapping

Apple MLX

API: mlx.nn.MaxPool2d
Strategy: Direct Mapping

Flax NNX

API: flax.nnx.max_pool
Strategy: Direct Mapping

PaxML / Praxis

API: praxis.layers.MaxPool2d
Strategy: Direct Mapping