评价此页

torch.Tensor.bernoulli#

Tensor.bernoulli(*, generator=None) Tensor#

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

请参阅 torch.bernoulli()