jax.random.maxwell

内容

jax.random.maxwell#

jax.random.maxwell(key, shape=(), dtype=<class 'float'>)[source]#

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

这些值服从以下概率密度函数的分布

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

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

参数:
  • key (KeyArrayLike) – 一个 PRNG 键。

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

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

返回值:

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

返回类型:

Array