OneHotOrdinal

類 torchrl.modules.OneHotOrdinal(scores: Tensor)

Ordinal 分佈的獨熱版本。

引數:

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