tensordict.nn.distributions.OneHotCategorical

class tensordict.nn.distributions.OneHotCategorical(logits: torch.Tensor | None = None, probs: torch.Tensor | None = None, **kwargs)

独热编码类别分布。

此类与 torch.distributions.Categorical 的行为完全相同，只是它读取和生成离散张量的独热编码。