Calculate log probability under negative binomial distribution.
Adapted from scvi-tools.
Usage
log_nb(x, mu, theta, eps = 1e-08)
Arguments
- x
Observed counts
- mu
Mean parameter
- theta
Dispersion parameter
- eps
Small constant for numerical stability
Value
Tensor of log probabilities
References
Gayoso et al. scvi-tools. https://github.com/YosefLab/scvi-tools