mindspore.ops.randint_like

mindspore.ops.randint_like(input, low, high, *, dtype=None, seed=None)

返回一个Tensor，其元素为 [ low , high ) 区间的随机整数，根据 input 决定shape和dtype。

警告

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

参数：

• input (Tensor) - 输入的Tensor，用来决定输出Tensor的shape和默认的dtype。

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

• high (int) - 随机区间的结束值。

• seed (int，可选) - 随机种子，必须大于或等于0。默认值： None ，值将取 0 。

关键字参数：

• dtype (mindspore.dtype，可选) - 指定的Tensor dtype，必须是int类型的dtype。如果是 None ，将会使用 input 的dtype。默认值： None 。

返回：

Tensor，shape和dtype被输入指定，其元素为 [ low , high ) 区间的随机整数。

异常：

• TypeError - 如果 seed 不是非负整数。

• TypeError - 如果 low 或 high 不是整数。

• ValueError - 如果 dtype 不是 mstype.int_type 类型。

支持平台：

Ascend GPU CPU

样例：

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