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 (KeyArrayLike) – 一个 PRNG 键。

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

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

返回:

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

返回类型:

Array