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