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)`` .. raw:: html

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