评价此页

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()