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)