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

Merge pull request #15283 from JiaYaobo:fix_wald_doc

Browse Source 
PiperOrigin-RevId: 520364879
This commit is contained in:
jax authors 2023-03-29 10:01:27 -07:00
commit cfa330b6fa
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):