torch.Tensor.bernoulli_

Tensor.bernoulli_(p=0.5, *, generator=None) → Tensor

Заполняет каждое местоположение self независимым образцом из $\text{Bernoulli}(\texttt{p})$. self может иметь целочисленные dtype.

p должно быть либо скаляром, либо тензором, содержащим вероятности, используемые для генерации бинарного случайного числа.

Если это тензор, то $\text{i}^{th}$ элемент тензора self будет установлен в значение, взятое из $\text{Bernoulli}(\texttt{p\_tensor[i]})$. В этом случае p должен иметь плавающие dtype.

См. также bernoulli() и torch.bernoulli()