Python PyTorch RelaxedBernoulli用法及代码示例

本文简要介绍python语言中 torch.distributions.relaxed_bernoulli.RelaxedBernoulli 的用法。

用法:

class torch.distributions.relaxed_bernoulli.RelaxedBernoulli(temperature, probs=None, logits=None, validate_args=None)

参数：

• temperature(Tensor) -松弛温度

• probs(数字,Tensor) -抽样概率1

• logits(数字,Tensor) -采样1的log-odds

基础：torch.distributions.transformed_distribution.TransformedDistribution

创建 RelaxedBernoulli 分布，由 temperature 以及 probs 或 logits (但不能同时使用两者)进行参数化。这是 Bernoulli 分布的宽松版本，因此值位于 (0, 1) 中，并且具有可重新参数化的样本。

例子：

>>> m = RelaxedBernoulli(torch.tensor([2.2]),
                         torch.tensor([0.1, 0.2, 0.3, 0.99]))
>>> m.sample()
tensor([ 0.2951,  0.3442,  0.8918,  0.9021])