jax.random.double_sided_maxwell

jax.random.double_sided_maxwell(key, loc, scale, shape=(), dtype=<class 'float'>)

从双侧麦克斯韦分布中采样。

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

\[f(x;\mu,\sigma) \propto z^2 e^{-z^2 / 2}\]

其中 \(z = (x - \mu) / \sigma\)，中心 \(\mu\) 由 loc 指定，尺度 \(\sigma\) 由 scale 指定。

参数:

• key (ArrayLike) – 一个 PRNG 键。

• loc (实数数组) – 分布的位置参数。

• scale (实数数组) – 分布的尺度参数。

• shape (形状) – 添加到参数 loc 和 scale 的可广播形状。

• dtype (浮点型DTypeLike) – 用于样本的类型。

返回:

一个 jnp.array 类型的样本数组。

返回类型:

数组