torch.Tensor.bernoulli#

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

Returns a result tensor where each$\mathtt{\text{result[i]}}$result[i] is independentlysampled from$\text{Bernoulli}(\mathtt{\text{self[i]}})$Bernoulli(self[i]).self must havefloating pointdtype, and the result will have the samedtype.

Seetorch.bernoulli()