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 编码。