Shortcuts

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.

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources