mindspore.Tensor.random_

查看源文件
mindspore.Tensor.random_(from_=0, to=None, *, generator=None)[源代码]

在区间 \([from\_, to-1]\) 内生成服从均匀分布的离散均匀分布随机数,原地更新输入tensor。

警告

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

参数:
  • from_ (Union[number.Number, Tensor], 可选) - 均匀分布的下界,可以是一个标量值或只有单个元素的任意维度的Tensor,默认值: 0

  • to (Union[number.Number, Tensor], 可选) - 均匀分布的上界,可以是一个标量值或只有单个元素的任意维度的Tensor。默认为输入数据类型的取值上限。默认值: None

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

返回:

返回输入tensor。

异常:
  • TypeError - from_to 不是整型。

  • RuntimeError - 如果 from_ 大于等于 to

支持平台:

Ascend

样例:

>>> from mindspore import Tensor
>>> a = Tensor([[2, 3, 4], [1, 2, 3]])
>>> from_ = 0
>>> to = 5
>>> print(a.random_(low, high).shape)
(2, 3)