cupy.random.geometric#

cupy.random.geometric(p, size=None, dtype=<class 'int'>)[source]#

几何分布。

返回从几何分布中抽取的样本数组。其概率质量函数定义为

\[f(x) = p(1-p)^{k-1}.\]
参数:
  • p (float) – 几何分布的成功概率。

  • size (inttuple of ints) – 数组的形状。如果为 None,则生成零维数组。

  • dtype – 数据类型指定符。只允许 numpy.int32numpy.int64 类型。

返回:

从几何分布中抽取的样本。

返回类型:

cupy.ndarray