GlobalAveragePooling2DΒΆ

Global average pooling operation for 2D data.

Abstract Signature:

GlobalAveragePooling2D(data_format: str, keepdims: bool = False)

PyTorch

API: torch.nn.AdaptiveAvgPool2d
Strategy: Direct Mapping

Keras

API: keras.layers.GlobalAveragePooling2D
Strategy: Direct Mapping

TensorFlow

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

Apple MLX

API: mlx.nn.AvgPool2d
Strategy: Direct Mapping

Flax NNX

API: flax.nnx.AvgPool
Strategy: Direct Mapping