tensordict.nn.distributions.OneHotCategorical

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

独热分类分布。

此类的行为与 torch.distributions.Categorical 完全一致，不同之处在于它读取和生成离散张量的独热编码。