mindspore.nn.probability

概率。

用于构建概率网络的高级组件。

Bayesian Layers

接口名	概述	支持平台
mindspore.nn.probability.bnn_layers.ConvReparam	具有重构化参数的卷积变分层。	Ascend GPU
mindspore.nn.probability.bnn_layers.DenseLocalReparam	具有局部重构化参数的密集变分层。	Ascend GPU
mindspore.nn.probability.bnn_layers.DenseReparam	具有重构参数化的密集变分层。	Ascend GPU

Prior and Posterior Distributions

接口名	概述	支持平台
mindspore.nn.probability.bnn_layers.NormalPosterior	用可训练的参数构建正态分布。	Ascend GPU
mindspore.nn.probability.bnn_layers.NormalPrior	初始化均值 0 和标准差 0.1 的正态分布。	Ascend GPU

Bayesian Wrapper Functions

接口名	概述	支持平台
mindspore.nn.probability.bnn_layers.WithBNNLossCell	为 BNN 生成一个合适的 WithLossCell，用损失函数包装贝叶斯网络。	Ascend GPU