OneHotOrdinal

class torchrl.modules.OneHotOrdinal(scores: Tensor)

Ordinal 分布的 one-hot 版本。

参数:

scores (torch.Tensor) – 一个形状为 […, N] 的张量，其中 N 是支持分布的集合的大小。通常是参数化分布的神经网络的输出。