torch.Tensor.bernoulli

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

返回一个结果张量，其中每个 $\texttt{result[i]}$ 都独立地从 $\text{Bernoulli}(\texttt{self[i]})$ 中采样。 self 必须是浮点数 dtype ，结果将具有相同的 dtype 。

查看 torch.bernoulli()