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\}$。