torch.Tensor.bernoulli_¶
- Tensor.bernoulli_(p=0.5, *, generator=None) Tensor ¶
用来自 的独立样本填充
self
的每个位置。self
可以具有整数dtype
。p
应该是一个标量或包含用于绘制二进制随机数的概率的张量。如果它是一个张量,则 元素的
self
张量将被设置为从 中采样的值。 在这种情况下,p 必须具有浮点dtype
。另请参见
bernoulli()
和torch.bernoulli()