mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove scale in wald docstring

Browse Source
This commit is contained in:
jiayaobo 2023-03-29 11:39:49 +08:00
parent c2d6fcc0e6
commit 3a4d0b3552
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
jax/_src/random.py
View File

@ -2069,7 +2069,7 @@ def wald(key: KeyArray,
Returns:
A random array with the specified dtype and with shape given by ``shape`` if
``shape`` is not None, or else by ``mean.shape`` and ``scale.shape``.
``shape`` is not None, or else by ``mean.shape``.
"""
key, _ = _check_prng_key(key)
if not dtypes.issubdtype(dtype, np.floating):