torch.Tensor.bernoulli¶

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

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