mindspore.ops.randint_like

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

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

警告

  • 2.9.0(不含)之后版本非兼容性变更:签名将调整为 randint_like(input, low=0, high, *, dtype=None)seed 参数将移除,low 改为可选(默认值 0 )。

  • 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]]