torch.bernoulli
时间: 2024-01-11 16:02:13 浏览: 12
浅谈pytorch torch.backends.cudnn设置作用
5星 · 资源好评率100%
torch.bernoulli是PyTorch中的一个函数,用于生成一个服从伯努利分布的张量。伯努利分布是一种二项分布,其随机变量只有两个取值,通常用0和1表示。在PyTorch中,torch.bernoulli可以指定生成的张量中1的概率,返回的张量中1的位置由随机生成器决定。如果随机生成器没有被指定,则使用默认的全局随机生成器。
阅读全文