jax.random.maxwell#
- jax.random.maxwell(key, shape=(), dtype=<class 'float'>)[source]#
Sample from a one sided Maxwell distribution.
The values are distributed according to the probability density function:
\[f(x) \propto x^2 e^{-x^2 / 2}\]on the domain \(0 \le x < \infty\).
- Parameters:
key (ArrayLike) – a PRNG key.
shape (Shape) – The shape of the returned samples.
dtype (DTypeLikeFloat) – The type used for samples.
- Returns:
A jnp.array of samples, of shape shape.
- Return type: