mindspore.mint.randint

mindspore.mint.randint(low, high, size, *, generator=None, dtype=None)

Returns a new tensor filled with integer numbers from the uniform distribution over an interval \([low, high)\) based on the given shape and dtype.

Warning

This is an experimental API that is subject to change or deletion.

Parameters

• low (int) – the lower bound of the generated random number

• high (int) – the upper bound of the generated random number

• size (Union[tuple(int), list(int)]) – Shape of the new tensor, e.g. \((2, 3)\).

Keyword Arguments

• generator (mindspore.Generator, optional) – a pseudorandom number generator. Default: None, uses the default pseudorandom number generator.

• dtype (mindspore.dtype, optional) – Designated tensor dtype. If None, mindspore.int64 will be applied. Default: None .

Returns

Tensor, with the designated shape and dtype, filled with random numbers from the uniform distribution on the interval \([low, high)\).

Raises

• TypeError – If size is not a tuple.

• TypeError – If low or high is not integer.

Supported Platforms:

Ascend

Examples

>>> from mindspore import mint
>>> print(mint.randint(0, 5, (2, 3)).shape)
(2, 3)