mindspore.ops.randint_like

View Source On Gitee
mindspore.ops.randint_like(input, low, high, seed=None, *, dtype=None)[source]

Returns a tensor with the same shape as input whose elements are random integers in the range of [ low , high ) .

Warning

The Ascend backend does not support the reproducibility of random numbers, so the seed parameter has no effect.

Parameters
  • input (Tensor) – The input tensor.

  • low (int) – Start value of interval.

  • high (int) – End value of interval.

  • seed (int, optional) – Random seed, must be non-negative. Default None .

Keyword Arguments

dtype (mindspore.dtype, optional) – The data type returned. Default None .

Returns

Tensor

Supported Platforms:

Ascend GPU CPU

Examples

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