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 样本。

返回类型:

数组