torch.Tensor.bernoulli_

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

用來自 $\text{Bernoulli}(\texttt{p})$ 的獨立樣本填充 self 中的每個位置。self 可以是整數 dtype。

p 應該是一個標量或一個包含用於抽取二元隨機數的機率的張量。

如果 p 是一個張量，self 張量的第 $\text{i}^{th}$ 元素將設定為從 $\text{Bernoulli}(\texttt{p\_tensor[i]})$ 中取樣得到的值。在這種情況下，p 必須具有浮點 dtype。

另請參閱 bernoulli() 和 torch.bernoulli()