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 CPU

mindspore.nn.probability.bijector.PowerTransform

PowerTransform 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

Categorical distribution.

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

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 CPU

mindspore.nn.probability.distribution.Uniform

Uniform Distribution.

Ascend GPU CPU