OneHotOrdinal¶
- class torchrl.modules.OneHotOrdinal(scores: Tensor)[source]¶
The one-hot version of the
Ordinal
distribution.- Parameters:
scores (torch.Tensor) – a tensor of shape […, N] where N is the size of the set which supports the distributions. Typically, the output of a neural network parametrising the distribution.