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