mindspore.nn.probability.bnn_layers.NormalPrior

class mindspore.nn.probability.bnn_layers.NormalPrior(dtype=mstype.float32, mean=0, std=0.1)[source]

To initialize a normal distribution of mean 0 and standard deviation 0.1.

Parameters

dtype (mindspore.dtype) – The argument is used to define the data type of the output tensor. Default: mindspore.float32.
mean (int, float) – Mean of normal distribution. Default: 0.
std (int, float) – Standard deviation of normal distribution. Default: 0.1.

Returns

Cell, a normal distribution.

Supported Platforms:: Ascend GPU