torch.Tensor.bernoulli

张量.bernoulli(*, generator=None) → 张量

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