mindspore.nn.probability

Probability.

The high-level components used to construct the probabilistic network.

Bijectors

API Name	Description	Supported Platforms
mindspore.nn.probability.bijector.Bijector	Bijecotr class.	Ascend GPU
mindspore.nn.probability.bijector.Exp	Exponential Bijector.	Ascend GPU
mindspore.nn.probability.bijector.GumbelCDF	GumbelCDF Bijector.	Ascend GPU
mindspore.nn.probability.bijector.Invert	Invert Bijector.	Ascend GPU
mindspore.nn.probability.bijector.PowerTransform	Power Bijector.	Ascend GPU
mindspore.nn.probability.bijector.ScalarAffine	Scalar Affine Bijector.	Ascend GPU
mindspore.nn.probability.bijector.Softplus	Softplus Bijector.	Ascend GPU

Bayesian Layers

API Name	Description	Supported Platforms
mindspore.nn.probability.bnn_layers.ConvReparam	Convolutional variational layers with Reparameterization.	Ascend GPU
mindspore.nn.probability.bnn_layers.DenseLocalReparam	Dense variational layers with Local Reparameterization.	Ascend GPU
mindspore.nn.probability.bnn_layers.DenseReparam	Dense variational layers with Reparameterization.	Ascend GPU

Prior and Posterior Distributions

API Name	Description	Supported Platforms
mindspore.nn.probability.bnn_layers.NormalPosterior	Build Normal distributions with trainable parameters.	Ascend GPU
mindspore.nn.probability.bnn_layers.NormalPrior	To initialize a normal distribution of mean 0 and standard deviation 0.1.	Ascend GPU

Bayesian Wrapper Functions

API Name	Description	Supported Platforms
mindspore.nn.probability.bnn_layers.WithBNNLossCell	Generate a suitable WithLossCell for BNN to wrap the bayesian network with loss function.	Ascend GPU

Distributions

API Name	Description	Supported Platforms
mindspore.nn.probability.distribution.Bernoulli	Bernoulli Distribution.	Ascend GPU
mindspore.nn.probability.distribution.Beta	Beta distribution.	Ascend
mindspore.nn.probability.distribution.Categorical	Create a categorical distribution parameterized by event probabilities.	Ascend GPU
mindspore.nn.probability.distribution.Cauchy	Cauchy distribution.	Ascend
mindspore.nn.probability.distribution.Distribution	Base class for all mathematical distributions.	Ascend GPU
mindspore.nn.probability.distribution.Exponential	Example class: Exponential Distribution.	Ascend GPU
mindspore.nn.probability.distribution.Gamma	Gamma distribution.	Ascend
mindspore.nn.probability.distribution.Geometric	Geometric Distribution.	Ascend GPU
mindspore.nn.probability.distribution.Gumbel	Gumbel distribution.	Ascend GPU
mindspore.nn.probability.distribution.Logistic	Logistic distribution.	Ascend GPU
mindspore.nn.probability.distribution.LogNormal	LogNormal distribution.	Ascend GPU
mindspore.nn.probability.distribution.Normal	Normal distribution.	Ascend GPU
mindspore.nn.probability.distribution.Poisson	Poisson Distribution.	Ascend
mindspore.nn.probability.distribution.TransformedDistribution	Transformed Distribution.	Ascend GPU
mindspore.nn.probability.distribution.Uniform	Example class: Uniform Distribution.	Ascend GPU

Deep Probability Networks

API Name	Description	Supported Platforms
mindspore.nn.probability.dpn.ConditionalVAE	Conditional Variational Auto-Encoder (CVAE).	Ascend GPU
mindspore.nn.probability.dpn.VAE	Variational Auto-Encoder (VAE).	Ascend GPU

Infer

API Name	Description	Supported Platforms
mindspore.nn.probability.infer.ELBO	The Evidence Lower Bound (ELBO).	Ascend GPU
mindspore.nn.probability.infer.SVI	Stochastic Variational Inference(SVI).	Ascend GPU

ToolBox

API Name	Description	Supported Platforms
mindspore.nn.probability.toolbox.UncertaintyEvaluation	Toolbox for Uncertainty Evaluation.	Ascend GPU
mindspore.nn.probability.toolbox.VAEAnomalyDetection	Toolbox for anomaly detection by using VAE.	Ascend GPU

Model Transformer

API Name	Description	Supported Platforms
mindspore.nn.probability.transforms.TransformToBNN	Transform Deep Neural Network (DNN) model to Bayesian Neural Network (BNN) model.	Ascend GPU