mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 12:26:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #23257 from carlosgmartin:random-orthogonal-citation

Browse Source 
PiperOrigin-RevId: 671156668
This commit is contained in:
jax authors 2024-09-04 17:24:51 -07:00
commit 720dfd7e43
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
jax/_src/random.py
View File

@ -2047,6 +2047,11 @@ def orthogonal(
Returns:
A random array of shape `(*shape, n, n)` and specified dtype.
References:
.. [1] Mezzadri, Francesco. (2007). "How to generate random matrices from
the classical compact groups". Notices of the American Mathematical
Society, 54(5), 592-604. https://arxiv.org/abs/math-ph/0609050.
"""
shape = core.canonicalize_shape(shape)
key, _ = _check_prng_key("orthogonal", key)