torch.Tensor.geometric_

Tensor.geometric_(p, *, generator=None) → Tensor

使用从几何分布中抽取的元素填充self张量：

$P(X=k) = (1 - p)^{k - 1} p, k = 1, 2, ...$

注意

torch.Tensor.geometric_() 中第 k 次试验是第一次成功，因此采样范围为 $\{1, 2, \ldots\}$；而torch.distributions.geometric.Geometric() 中第 $(k+1)$ 次试验是第一次成功，因此采样范围为 $\{0, 1, \ldots\}$。