jax.random.双侧麦克斯韦分布

jax.random.双侧麦克斯韦分布#

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

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

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

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

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

返回:

样本的 jnp.array。

返回类型:

Array