torch.Tensor.geometric_

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

Fills self tensor with elements drawn from the geometric distribution:

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