Python PyTorch Geometric用法及代碼示例

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

用法:

class torch.distributions.geometric.Geometric(probs=None, logits=None, validate_args=None)

參數：

• probs(數字,Tensor) -抽樣的概率 1 。必須在範圍內 (0, 1]

• logits(數字,Tensor) -采樣 1 的 log-odds 。

基礎：torch.distributions.distribution.Distribution

創建由 probs 參數化的幾何分布，其中 probs 是伯努利試驗的成功概率。它表示在 k + 1 伯努利試驗中，第一個 k 試驗在成功之前失敗的概率。

樣本是非負整數 [0, \inf )。

例子：

>>> m = Geometric(torch.tensor([0.3]))
>>> m.sample()  # underlying Bernoulli has 30% chance 1; 70% chance 0
tensor([ 2.])