torch.Tensor.geometric_

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

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

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

注意

torch.Tensor.geometric_() k-th trial 是第一次成功，因此从 $\{1, 2, \ldots\}$ 中抽取样本，而 torch.distributions.geometric.Geometric() $(k+1)$-th trial 是第一次成功 因此从 $\{0, 1, \ldots\}$ 中抽取样本。