Python PyTorch RelaxedOneHotCategorical用法及代碼示例

本文簡要介紹python語言中 torch.distributions.relaxed_categorical.RelaxedOneHotCategorical 的用法。

用法:

class torch.distributions.relaxed_categorical.RelaxedOneHotCategorical(temperature, probs=None, logits=None, validate_args=None)

參數：

• temperature(Tensor) -鬆弛溫度

• probs(Tensor) -事件概率

• logits(Tensor) -每個事件的非標準化對數概率

基礎：torch.distributions.transformed_distribution.TransformedDistribution

創建由 temperature 以及 probs 或 logits 參數化的 RelaxedOneHotCategorical 分布。這是 OneHotCategorical 發行版的寬鬆版本，因此它的樣本是單純形的，並且是可重新參數化的。

例子：

>>> m = RelaxedOneHotCategorical(torch.tensor([2.2]),
                                 torch.tensor([0.1, 0.2, 0.3, 0.4]))
>>> m.sample()
tensor([ 0.1294,  0.2324,  0.3859,  0.2523])