jax.random.double_sided_maxwellΒΆ

jax.random.double_sided_maxwell(key, loc, scale, shape=(), dtype=<class 'numpy.float64'>)[source]ΒΆ

Sample from a double sided Maxwell distribution.

Samples using:

loc + scale* sgn(U-0.5)* one_sided_maxwell U~Unif;

Parameters
  • key (ndarray) – a PRNGKey key.

  • loc (Any) – The location parameter of the distribution.

  • scale (Any) – The scale parameter of the distribution.

  • shape (Sequence[int]) – The shape added to the parameters loc and scale broadcastable shape.

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

Return type

ndarray

Returns

A jnp.array of samples.