cyjax.random.uniform_angle#
- cyjax.random.uniform_angle(key, shape, minrad=0, maxrad=1)#
Complex values individually uniform on complex disk.
Radius of complex valuse lie between
minrad
(default 0, corresponding to a disk) andmaxrad
.Example
>>> x = uniform_angle(jax.random.PRNGKey(0), (4,)) >>> jnp.all(jnp.abs(x) < 1) DeviceArray(True, dtype=bool)
- Parameters:
key (
Union
[PRNGKeyArray
,Array
]) – PRNG random key.shape (
Sequence
[Union
[int
,Any
]]) – Tuple of nonnegative integers.minrad (
Union
[float
,int
]) – Minimal absolute value of generated values.maxrad (
Union
[float
,int
]) – Maximal absolute value of generated values.
- Return type:
Union
[Array
,ndarray
,bool_
,number
]- Returns:
Array of requested
shape
containing complex values.