Python PyTorch OneHotCategorical用法及代码示例

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

用法:

class torch.distributions.one_hot_categorical.OneHotCategorical(probs=None, logits=None, validate_args=None)

参数：

• probs(Tensor) -事件概率

• logits(Tensor) -事件日志概率(未标准化)

基础：torch.distributions.distribution.Distribution

创建由 probs 或 logits 参数化的 one-hot 分类分布。

样本是 one-hot 大小为 probs.size(-1) 的编码向量。

注意

probs 参数必须是非负、有限且具有非零总和，并且沿最后一个维度将其归一化为总和为 1。 probs 将返回此标准化值。 logits 参数将被解释为非标准化对数概率，因此可以是任何实数。它同样会被标准化，以便沿最后一个维度得到的概率总和为 1。 logits 将返回此标准化值。

另请参阅：torch.distributions.Categorical() 了解 probs 和 logits 的规范。

例子：

>>> m = OneHotCategorical(torch.tensor([ 0.25, 0.25, 0.25, 0.25 ]))
>>> m.sample()  # equal probability of 0, 1, 2, 3
tensor([ 0.,  0.,  0.,  1.])