torch.Tensor.bernoulli_

Tensor.bernoulli_(p=0.5, *, generator=None) → Tensor

用独立的样本填充 self 的每个位置，样本来自 $\text{Bernoulli}(\texttt{p})$。 self 可以具有整数 dtype。

p 应该是一个标量或张量，包含用于绘制二元随机数的概率。

如果它是一个张量，则 $\text{i}^{th}$ 元素的 self 张量将被设置为从 $\text{Bernoulli}(\texttt{p\_tensor[i]})$ 采样的值。在这种情况下，p 必须具有浮点 dtype。

另请参阅 bernoulli() 和 torch.bernoulli()