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:
AscendGPU