OneHotOrdinal

class torchrl.modules.OneHotOrdinal(scores: Tensor)

Ordinal 分佈的 one-hot 版本。

引數:

scores (torch.Tensor) – 一個形狀為 […, N] 的張量，其中 N 是支援該分佈的集合的大小。通常是引數化該分佈的神經網路的輸出。