評價此頁

torch.Tensor.bernoulli#

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

返回一個結果張量,其中每個 result[i]\texttt{result[i]} 獨立地從 Bernoulli(self[i])\text{Bernoulli}(\texttt{self[i]}) 中取樣。 `self` 必須具有浮點 `dtype`,結果也將具有相同的 `dtype`。

請參閱 torch.bernoulli()