torch.Tensor.bernoulli

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

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

См. torch.bernoulli()