zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-08-20 18:39:45 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update normalization.py

Browse Source
This commit is contained in:
__mo_san__ 2023-12-22 20:57:32 +01:00 committed by m0saan
parent a43b853194
commit b444a6a693
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python/mlx/nn/layers/normalization.py
View File

@ -208,7 +208,7 @@ class BatchNorm(Module):
>>> mx.random.seed(42)
>>> input = mx.random.normal((5, 4), dtype=mx.float32)
>>> # Batch norm
>>> bn = nn.BatchNorm1d(num_features=4, affine=True)
>>> bn = nn.BatchNorm(num_features=4, affine=True)
>>> output = bn(x)
"""