Probability Distributions
=========================

This module contains sparse multivariate normal distributions.

Sparse Multivariate Normal
---------------------------

.. currentmodule:: torchsparsegradutils.distributions.sparse_multivariate_normal

.. autoclass:: SparseMultivariateNormal
   :members:
   :undoc-members:
   :show-inheritance:
   :exclude-members: has_rsample, diagonal, scale_tril, precision_tril, loc

.. autoclass:: SparseMultivariateNormalNative
   :members:
   :undoc-members:
   :show-inheritance:
   :exclude-members: scale_tril, loc

Constraints
-----------

.. currentmodule:: torchsparsegradutils.distributions.constraints

.. automodule:: torchsparsegradutils.distributions.constraints
   :members:
   :undoc-members:
   :show-inheritance: