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.])