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
。
- 关键字参数:
dtype (
mindspore.dtype
,可选) - 指定数据类型。默认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]]