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 | TypedNdArray) – 数组,计算 PMF 的值 p (Array | ndarray | bool | number | bool | int | float | complex | TypedNdArray) – 数组,分布形状参数 loc (Array | ndarray | bool | number | bool | int | float | complex | TypedNdArray) – 数组,分布偏移参数 返回: logpmf 值的数组。 返回类型: Array 另请参阅 jax.scipy.stats.geom.pmf()