tensordict.nn.distributions.OneHotCategorical

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

One-hot 分類分佈。

這個類別的行為與 torch.distributions.Categorical 完全相同，除了它讀取和產生離散張量的一次性編碼。