torch.Tensor.geometric_

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

Заполняет тензор self элементами, взятыми из геометрического распределения:

$$f(X=k) = p^{k - 1} (1 - p)$$