jax.random.maxwell

jax.random.maxwell(key, shape=(), dtype=None)

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

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

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

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

参数:

• key (ArrayLike) – 一个 PRNG 密钥。

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

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

返回:

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

返回类型:

Array