jax.random.rademacherΒΆ

jax.random.rademacher(key, shape, dtype=<class 'numpy.int64'>)[source]ΒΆ

Sample from a Rademacher distribution.

Parameters
  • key (ndarray) – a PRNGKey key.

  • shape (Sequence[int]) – The shape of the returned samples.

  • dtype (Any) – The type used for samples.

Return type

ndarray

Returns

A jnp.array of samples, of shape shape. Each element in the output has a 50% change of being 1 or -1.