avgpool, not maxpool (#1002)

Author: a1eaiactaest

python/mlx/nn/layers/pooling.py

@@ -317,7 +317,7 @@ class AvgPool2d(_Pool2d):
         >>> import mlx.core as mx
         >>> import mlx.nn.layers as nn
         >>> x = mx.random.normal(shape=(8, 32, 32, 4))
-        >>> pool = nn.MaxPool2d(kernel_size=2, stride=2)
+        >>> pool = nn.AvgPool2d(kernel_size=2, stride=2)
         >>> pool(x)
     """