Python PyTorch Binomial用法及代碼示例

本文簡要介紹python語言中 torch.distributions.binomial.Binomial 的用法。

用法:

class torch.distributions.binomial.Binomial(total_count=1, probs=None, logits=None, validate_args=None)

參數：

• total_count(int或者Tensor) -伯努利試驗次數

• probs(Tensor) -事件概率

• logits(Tensor) -事件log-odds

基礎：torch.distributions.distribution.Distribution

創建由 total_count 和 probs 或 logits(但不是兩者)參數化的二項式分布。 total_count 必須可與 probs /logits 一起廣播。

例子：

>>> m = Binomial(100, torch.tensor([0 , .2, .8, 1]))
>>> x = m.sample()
tensor([   0.,   22.,   71.,  100.])

>>> m = Binomial(torch.tensor([[5.], [10.]]), torch.tensor([0.5, 0.8]))
>>> x = m.sample()
tensor([[ 4.,  5.],
        [ 7.,  6.]])