mindspore.ops.randint_like

查看源文件
mindspore.ops.randint_like(input, low, high, *, dtype=None, seed=None)[源代码]

返回一个与输入shape相同的tensor,其元素为 [ low , high ) 区间的随机整数。

警告

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

参数:
  • input (Tensor) - 输入tensor。

  • low (int) - 随机区间的起始值。

  • high (int) - 随机区间的末尾值。

  • seed (int,可选) - 随机种子,必须大于或等于0。默认 None

关键字参数:
返回:

Tensor

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> a = mindspore.tensor([[1, 2, 3], [3, 2, 1]])
>>> print(mindspore.ops.randint_like(a, 1, 10))
[[4 9 7]
[9 1 2]]