tensordict.nn.distributions.OneHotCategorical¶ class tensordict.nn.distributions.OneHotCategorical(logits: Optional[Tensor] = None, probs: Optional[Tensor] = None, **kwargs)¶ 独热(One-hot)分类分布。 此类行为与 torch.distributions.Categorical 完全相同,只不过它读取和生成离散张量的独热编码。