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])