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