torch.Tensor.bernoulli

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

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

См. torch.bernoulli()