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

Fix docs: replace mx.random.randn with mx.random.normal (#2890)

Browse Source
This commit is contained in:
Satyam singh
2025-12-10 01:16:30 +05:30
committed by GitHub
parent 27232db1ba
commit f7a400951a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docs/src/usage/indexing.rst
View File

@@ -186,7 +186,7 @@ Boolean masks follow NumPy semantics:
.. code-block:: shell .. code-block:: shell
>>> a = mx.arange(1000).reshape(10, 10, 10) >>> a = mx.arange(1000).reshape(10, 10, 10)
>>> a[mx.random.randn(10, 10) > 0.0] = 0 # valid: mask covers axes 0 and 1 >>> a[mx.random.normal((10, 10)) > 0.0] = 0 # valid: mask covers axes 0 and 1
The mask of shape ``(10, 10)`` applies to the first two axes, so ``a[mask]`` The mask of shape ``(10, 10)`` applies to the first two axes, so ``a[mask]``
selects the 1-D slices ``a[i, j, :]`` where ``mask[i, j]`` is ``True``. selects the 1-D slices ``a[i, j, :]`` where ``mask[i, j]`` is ``True``.