MindSpore Probability Python API
================================

.. automodule:: mindspore.nn.probability

Bayesian Layers
---------------

.. msplatformautosummary::
    :toctree: nn_probability
    :nosignatures:
    :template: classtemplate_probability.rst

    mindspore.nn.probability.bnn_layers.ConvReparam
    mindspore.nn.probability.bnn_layers.DenseLocalReparam
    mindspore.nn.probability.bnn_layers.DenseReparam

Prior and Posterior Distributions
----------------------------------

.. msplatformautosummary::
    :toctree: nn_probability
    :nosignatures:
    :template: classtemplate_probability.rst

    mindspore.nn.probability.bnn_layers.NormalPosterior
    mindspore.nn.probability.bnn_layers.NormalPrior

Bayesian Wrapper Functions
---------------------------

.. msplatformautosummary::
    :toctree: nn_probability
    :nosignatures:
    :template: classtemplate_probability.rst

    mindspore.nn.probability.bnn_layers.WithBNNLossCell

Deep Probability Networks
--------------------------

.. msplatformautosummary::
    :toctree: nn_probability
    :nosignatures:
    :template: classtemplate_probability.rst

    mindspore.nn.probability.dpn.ConditionalVAE
    mindspore.nn.probability.dpn.VAE

Infer
------

.. msplatformautosummary::
    :toctree: nn_probability
    :nosignatures:
    :template: classtemplate_probability.rst

    mindspore.nn.probability.infer.ELBO
    mindspore.nn.probability.infer.SVI

ToolBox
---------

.. msplatformautosummary::
    :toctree: nn_probability
    :nosignatures:
    :template: classtemplate_probability.rst

    mindspore.nn.probability.toolbox.UncertaintyEvaluation
    mindspore.nn.probability.toolbox.VAEAnomalyDetection

Model Transformer
------------------

.. msplatformautosummary::
    :toctree: nn_probability
    :nosignatures:
    :template: classtemplate_probability.rst

    mindspore.nn.probability.transforms.TransformToBNN