Shortcuts

torch.Tensor.bernoulli

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

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

参见 torch.bernoulli()