torch.Tensor.bernoulli_

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

用独立样本填充 self 的每个位置。 self 可以是整数 dtype 。

p 应为一个标量或包含用于抽取二进制随机数的概率的 tensor。

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

参见 bernoulli() 和 torch.bernoulli()