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