torch.bernoulli 的使用方法(附代码示例)
torch.bernoulli 的使用方法(附代码示例)功能解释参数代码示例功能解释*torch.bernoulli(input, , generator=None, out=None) → Tensor从伯努利分布中提取二进制随机数(0或1),输入张量应为包含用于绘制二进制随机数的概率的张量。因此,输入中的所有值都必须在以下范围内(0,1)。第i个元素的输出tensor将要根据输入的第i个概率值,来决定是否生成1值。输出tensor仅仅只有0或1值,并且其形状和输入的Input的形状是一样的。
原创
2020-08-06 10:11:31 ·
16854 阅读 ·
0 评论