mindspore.Tensor.normal_

查看源文件
mindspore.Tensor.normal_(mean=0, std=1, *, generator=None)[源代码]

使用随机数原地更新tensor,且随机数的采样服从由参数 meanstd 所构成的正态分布。

警告

这是一个实验性API,后续可能修改或删除。

参数:
  • mean (number, 可选) - 正态分布的均值,默认值: 0

  • std (number, 可选) - 正态分布的标准差,默认值: 1

关键字参数:
  • generator (mindspore.Generator, 可选) - 伪随机数生成器。默认值: None ,使用默认伪随机数生成器。

返回:

返回一个tensor,该Tensor由服从正态分布的随机数填充,且type和shape与原tensor一致。

异常:
  • TypeError - meanstd 的dtype不是number,即数据类型不是bool、int、float、complex之一。

支持平台:

Ascend

样例:

>>> import mindspore
>>> import numpy as np
>>> x = mindspore.Tensor(np.array([[1, 2], [3, 4]]), dtype=mindspore.float32)
>>> output = x.normal_()
>>> print(output)
[[0.2788825 1.3305743]
 [1.244194 1.16303174]]