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()