mxnet.gluon.probability.distributions.geometric

Geometric distribution class.

Classes

Geometric([prob, logit, validate_args])

Create a geometric distribution object.

class mxnet.gluon.probability.distributions.geometric.Geometric(prob=None, logit=None, validate_args=None)

Bases: Distribution

Create a geometric distribution object.

Parameters:

• prob (Tensor or scalar, default None) – Probability of sampling 1.

• logit (Tensor or scalar, default None) – The log-odds of sampling 1.

broadcast_to(batch_shape)

Returns a new distribution instance with parameters expanded to batch_shape. This method calls numpy.broadcast_to on the parameters.

Parameters:

batch_shape (Tuple) – The batch shape of the desired distribution.

entropy()

Returns entropy of distribution.

log_prob(value)

Returns the log of the probability density/mass function evaluated at value.

logit()

Get the log-odds of sampling 1.

Returns:

Parameter tensor.

Return type:

Tensor

property mean

Returns the mean of the distribution.

prob()

Get the probability of sampling 1.

Returns:

Parameter tensor.

Return type:

Tensor

sample(size=None)

Generates a shape shaped sample.

sample_n(size=None)

Generate samples of (n + parameter_shape) from the distribution.

property variance

Returns the variance of the distribution.