AvgPool

Pools the input by taking the average over a window.

Abstract Signature:

AvgPool(inputs: Tensor, window_shape: Tuple[int, ...], strides: Tuple[int, ...], padding: str = VALID)

PyTorch

API: torch.nn.functional.avg_pool2d
Strategy: Direct Mapping
Official Docs ↗

Keras

API: keras.ops.average_pool
Strategy: Direct Mapping
Official Docs ↗

TensorFlow

API: tf.nn.avg_pool
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: flax.nnx.avg_pool
Strategy: Direct Mapping
Official Docs ↗