jax.scipy.stats.geom.logpmf

jax.scipy.stats.geom.logpmf(k, p, loc=0)

几何对数概率质量函数。

JAX 实现 scipy.stats.geom logpmf。

几何概率质量函数由下式给出

\[f(k) = (1 - p)^{k-1}p\]

对于 \(k\ge 1\) 和 \(0 \le p \le 1\)。

参数:

• k (Array | ndarray | bool | number | bool | int | float | complex) – 类数组，要评估 PMF 的值

• p (Array | ndarray | bool | number | bool | int | float | complex) – 数组类型，分布形状参数

• loc (Array | ndarray | bool | number | bool | int | float | complex) – 数组类型，分布偏移参数

返回值:

对数概率质量函数值的数组。

返回值类型:

数组

参见

jax.scipy.stats.geom.pmf()