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

从给定区间生成随机整数。

值从半开区间 [low, high) 中的整数中以相等概率采样得到。下界和上界可以是标量或数组，并且必须能够广播到 shape。

参数:

• low (标量 或 array) – 区间的下界。

• high (标量 或 array) – 区间的上界。

• shape (list(int), 可选) – 输出的形状。默认值：()。

• dtype (Dtype, 可选) – 输出的类型。默认值：int32。

• key (array, 可选) – 一个 PRNG 键。默认值：None。

返回值:

随机整数数组。

返回类型:

array