torch.bernoulli()
伯努利分布是一个离散分布,有两个结果,即成功和失败。如果成功的概率是p,那么失败的概率是(1-p),反之亦然。...0.0966, 0.7385, 0.6546],
[0.4255, 0.8294, 0.8315],
[0.8065, 0.8228, 0.6467]])
现在我们把bernoulli...()函数应用到张量上
torch.bernoulli(a)
输出如下:
tensor([[0., 1., 1.],
[1., 1., 0.],
[1., 0.,