torch.Tensor.bernoulli_¶
- Tensor.bernoulli_(p=0.5, *, generator=None) Tensor ¶
Fills each location of
self
with an independent sample from .self
can have integraldtype
.p
should either be a scalar or tensor containing probabilities to be used for drawing the binary random number.If it is a tensor, the element of
self
tensor will be set to a value sampled from . In this case p must have floating pointdtype
.See also
bernoulli()
andtorch.bernoulli()