jax.random.randint#

jax.random.randint(key, shape, minval, maxval, dtype=<class 'int'>)[原始碼]#

使用給定的形狀/dtype 在 [minval, maxval) 範圍內取樣均勻隨機值。

參數:

  • key (ArrayLike) – 作為隨機金鑰使用的 PRNG 金鑰。

  • shape (Shape) – 代表形狀的非負整數元組。

  • minval (IntegerArray) – 與 shape 廣播相容的 int 或 int 陣列，範圍的最小值 (包含)。

  • maxval (IntegerArray) – 與 shape 廣播相容的 int 或 int 陣列，範圍的最大值 (不包含)。

  • dtype (DTypeLikeInt) – 選項，用於傳回值的 int dtype (如果 jax_enable_x64 為 true，則預設為 int64，否則為 int32)。

傳回:

具有指定形狀和 dtype 的隨機陣列。

傳回型別:

Array