jax.random.maxwell

jax.random.maxwell(key, shape=(), dtype=<class 'float'>)

从单边麦克斯韦分布中采样。

这些值根据概率密度函数分布

\[f(x) \propto x^2 e^{-x^2 / 2}\]

在定义域 \(0 \le x < \infty\) 上。

参数:

• key (ArrayLike) – 一个PRNG键。

• shape (Shape) – 返回样本的形状。

• dtype (DTypeLikeFloat) – 用于样本的类型。

返回:

一个形状为 shape 的样本的 jnp.array。

返回类型:

Array