mindspore.ops.normal

查看源文件
mindspore.ops.normal(shape, mean, stddev, seed=None)[源代码]

返回符合正态(高斯)分布的随机tensor。

警告

Ascend后端不支持随机数重现功能, seed 参数不起作用。

参数:
  • shape (tuple) - 返回tensor的shape。

  • mean (Union[Tensor, int, float]) - 返回tensor的正态分布平均值。

  • stddev (Union[Tensor, int, float]) - 返回tensor的正态分布标准差。

  • seed (int,可选) - 随机种子,默认 None ,等同于 0

返回:

Tensor

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> shape = (3, 1, 2)
>>> mean = mindspore.tensor([[3, 4], [5, 6]], mindspore.float32)
>>> stddev = mindspore.tensor(1.0, mindspore.float32)
>>> output = mindspore.ops.normal(shape, mean, stddev, seed=5)
>>> result = output.shape
>>> print(result)
(3, 2, 2)
>>> shape = (3, 1, 3)
>>> mean = mindspore.tensor([[3, 4, 3], [3, 5, 6]], mindspore.float32)
>>> stddev = mindspore.tensor(1.0, mindspore.float32)
>>> output = mindspore.ops.normal(shape, mean, stddev, seed=5)
>>> result = output.shape
>>> print(result)
(3, 2, 3)
>>> shape = (3, 1, 3)
>>> mean = mindspore.tensor([[1, 2, 3], [3, 4, 3], [3, 5, 6]], mindspore.float32)
>>> stddev = mindspore.tensor(1.0, mindspore.float32)
>>> output = mindspore.ops.normal(shape, mean, stddev, seed=5)
>>> result = output.shape
>>> print(result)
(3, 3, 3)