torch.Tensor.bernoulli¶

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

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