AvgPool2DΒΆ

Alias for AveragePooling2D.

Abstract Signature:

AvgPool2D(pool_size: Union[int, Tuple[int, int]], strides: Union[int, Tuple[int, int]], padding: str = valid, data_format: str)

PyTorch

API: torch.nn.AvgPool2d
Strategy: Direct Mapping

JAX (Core)

API: β€”
Strategy: Custom / Partial

Keras

API: keras.layers.AvgPool2D
Strategy: Direct Mapping

TensorFlow

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

Apple MLX

API: mlx.nn.AvgPool2d
Strategy: Direct Mapping

Flax NNX

API: flax.nnx.AvgPool
Strategy: Direct Mapping

PaxML / Praxis

API: praxis.layers.AvgPooling
Strategy: Direct Mapping