torch.Tensor.bernoulli¶

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

返回一個結果張量，其中每個 $\texttt{result[i]}$ 都獨立地從 $\text{Bernoulli}(\texttt{self[i]})$ 中取樣。self 必須具有浮點 dtype，並且結果將具有相同的 dtype。