GlobalPoolingΒΆ

Performs global pooling over the input.

Abstract Signature:

GlobalPooling(pooling_type: str = AVG, 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

PaxML / Praxis

API: paxml.layers.GlobalPooling
Strategy: Direct Mapping