torch.Tensor.bernoulli¶

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

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