tensordict.nn.distributions.OneHotCategorical

class tensordict.nn.distributions.OneHotCategorical(logits: Optional[Tensor] = None, probs: Optional[Tensor] = None, **kwargs)

獨熱（One-hot）分類分佈。

此類行為與 torch.distributions.Categorical 完全相同，只不過它讀取和生成離散張量的獨熱編碼。