OneHotOrdinal

class torchrl.modules.OneHotOrdinal(scores: Tensor)

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.