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