cyjax.random.uniform_angle_unit#

cyjax.random.uniform_angle_unit(key, shape)#

Complex values with unit modulus and uniform angle.

Example

>>> x = uniform_angle_unit(jax.random.PRNGKey(0), (4,))
>>> jnp.allclose(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.

Returns:

Array of requested shape containing complex values.