zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Formatting fixes (#1606)

Browse Source
This commit is contained in:
Angelos Katharopoulos
2024-11-20 15:30:36 -08:00
committed by GitHub
parent cb431dfc9f
commit d8c824c594
2 changed files with 25 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
python/mlx/nn/layers/pooling.py
View File

@@ -450,6 +450,7 @@ class AvgPool3d(_Pool3d):
>>> pool = nn.AvgPool3d(kernel_size=2, stride=2)
>>> pool(x)
"""
def __init__(
self,
kernel_size: Union[int, Tuple[int, int, int]],