.. currentmodule:: torchrl.modules

Distribution Classes
====================

Custom distribution classes for RL, extending PyTorch distributions.

.. autosummary::
    :toctree: generated/
    :template: rl_template_noinherit.rst

    Delta
    IndependentNormal
    MaskedCategorical
    NormalParamExtractor
    OneHotCategorical
    ReparamGradientStrategy
    TanhDelta
    TanhNormal
    TruncatedNormal