mindspore.mint.normal

mindspore.mint.normal(mean=0.0, std=1.0, size=None, generator=None)[源代码]

根据正态(高斯)随机数分布生成随机数。

参数:
  • mean (Union[float, Tensor],可选) - 每个元素的均值, shape与std相同。默认值: 0.0

  • std (Union[float, Tensor],可选) - 每个元素的标准差, shape与mean相同。std的值大于等于0。默认值: 1.0

  • size (tuple,可选) - 当mean和std为常量时,指定输出shape。默认值: None

  • generator (generator,可选) - MindSpore随机种子。默认值: None

返回:

Tensor,输出tensor的shape和mean的shape相同,或者在mean和std为常量时,shape为size。

异常:
  • TypeError - 如果 meanstd 不是Union[float, Tensor]。

支持平台:

Ascend

样例:

>>> import mindspore
>>> import numpy as np
>>> from mindspore import mint
>>> from mindspore import Tensor
>>> mean = Tensor(np.array([1.0, 2.0, 3.0]), mindspore.float32)
>>> std = Tensor(np.array([1.0, 2.0, 3.0]), mindspore.float32)
>>> output = mint.normal(mean, std)
>>> print(output.shape)
(3,)