tensordict.nn.distributions.OneHotCategorical

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

独热（One-hot）分类分布。

这个类与 torch.distributions.Categorical 的行为完全相同，但它读取和生成离散张量的独热（one-hot）编码。