快捷方式

torch.Tensor.bernoulli_

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

用来自 Bernoulli(p)\text{Bernoulli}(\texttt{p}) 的独立样本填充 self 的每个位置。 self 可以具有整数 dtype

p 应该是一个标量或包含用于绘制二进制随机数的概率的张量。

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

另请参见 bernoulli()torch.bernoulli()

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源