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

Add default shape to random.rademacher.

Browse Source
This commit is contained in:
carlosgmartin 2024-06-04 00:16:58 -04:00
parent 95ed39bfc0
commit de8a0d3be6
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
jax/_src/random.py
View File

@ -1860,7 +1860,7 @@ def _f(key, dfnum, dfden, shape, dtype) -> Array:
def rademacher(key: KeyArrayLike,
shape: Shape,
shape: Shape = (),
dtype: DTypeLikeInt = int) -> Array:
r"""Sample from a Rademacher distribution.
@ -1873,7 +1873,7 @@ def rademacher(key: KeyArrayLike,
Args:
key: a PRNG key.
shape: The shape of the returned samples.
shape: The shape of the returned samples. Default ().
dtype: The type used for samples.
Returns: