Negative binomial distribution.
One of the following parameterizations must be provided:
(total_count, probs) where total_count is the number of failures until
the experiment is stopped and probs the success probability.
- The (mu, theta) parameterization is the one used by scvi-tools. These parameters respectively
control the mean and inverse dispersion of the distribution.
Number of failures until the experiment is stopped.
The success probability.
Mean of the distribution.
Raise ValueError if arguments do not match constraints
Returns the shape over which parameters are batched.
Returns the shape of a single sample (without batching).
Returns the mean of the distribution.
Returns the standard deviation of the distribution.
Returns the variance of the distribution.
Returns the cumulative density/mass function evaluated at value.
Returns entropy of distribution, batched over batch_shape.
Returns tensor containing all values supported by a discrete distribution.
Returns a new distribution instance (or populates an existing instance provided by a derived class) with batch dimensions expanded to batch_shape.
Returns the inverse cumulative density/mass function evaluated at value.
Returns the log of the probability density/mass function evaluated at value.
Returns perplexity of distribution, batched over batch_shape.
Generates a sample_shape shaped reparameterized sample or sample_shape shaped batch of reparameterized samples if the distribution parameters are batched.
Generates a sample_shape shaped sample or sample_shape shaped batch of samples if the distribution parameters are batched.
Generates n samples or n batches of samples if the distribution parameters are batched.