mlx.core.random.randint

randint(low: scalar | array, high: scalar | array, shape: Sequence[int] = [], dtype: Dtype | None = int32, key: array | None = None, stream: None | Stream | Device = None) → array

從指定區間產生隨機整數。

The values are sampled with equal probability from the integers in half-open interval [low, high). The lower and upper bound can be scalars or arrays and must be broadcastable to shape.

參數:

• low (scalar or array) -- Lower bound of the interval.

• high (scalar or array) -- Upper bound of the interval.

• shape (list(int), optional) -- Shape of the output. Default: ().

• dtype (Dtype, optional) -- Type of the output. Default: int32.

• key (array, optional) -- A PRNG key. Default: None.

回傳:

The array of random integers.

回傳型別:

array