Clarified ambiguous docstring in NegativeBinomial

Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/25923

Differential Revision: D17392848

Pulled By: soumith

fbshipit-source-id: 2833e72fe449c74dfd8273a7b1eb46c05c63d999
This commit is contained in:
Kyle Engel
2019-09-16 14:30:12 -07:00
committed by Facebook Github Bot
parent 327e94f51b
commit 4a947b607c

View File

@ -8,8 +8,8 @@ from torch.distributions.utils import broadcast_all, probs_to_logits, lazy_prope
class NegativeBinomial(Distribution):
r"""
Creates a Negative Binomial distribution, i.e. distribution
of the number of independent identical Bernoulli trials
needed before :attr:`total_count` failures are achieved. The probability
of the number of successful independent and identical Bernoulli trials
before :attr:`total_count` failures are achieved. The probability
of success of each Bernoulli trial is :attr:`probs`.
Args: