torch.Tensor.bernoulli

Tensor.bernoulli(*, generator=None) → Tensor

Возвращает тензор результатов, где каждый $\texttt{result[i]}$ независимо выбирается из $\text{Bernoulli}(\texttt{self[i]})$. self должен иметь тип с плавающей точкой dtype, и результат будет иметь тот же dtype.

См. torch.bernoulli()