mindspore.nn.probability

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

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