tensordict.nn.distributions.OneHotCategorical

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

獨熱分類分佈。

此類的行為與 torch.distributions.Categorical 完全一致，不同之處在於它讀取和生成離散張量的獨熱編碼。