cupy.random.geometric¶

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

Geometric distribution.

Returns an array of samples drawn from the geometric distribution. Its probability mass function is defined as

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

Parameters:	p (float) – Success probability of the geometric distribution. size (int or tuple of ints) – The shape of the array. If `None`, a zero-dimensional array is generated. dtype – Data type specifier. Only `numpy.int32` and `numpy.int64` types are allowed.
Returns:	Samples drawn from the geometric distribution.
Return type:	cupy.ndarray