jax.random.uniform#

jax.random.uniform(key, shape=(), dtype=<class 'float'>, minval=0.0, maxval=1.0)[原始碼]#

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

參數:

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

  • shape (Shape) – 選填，代表結果形狀的非負整數元組。預設值為 ()。

  • dtype (DTypeLikeFloat) – 選填，用於傳回值的浮點數 dtype (如果 jax_enable_x64 為 true，則預設值為 float64，否則為 float32)。

  • minval (RealArray) – 選填，範圍的最小值 (包含)，與形狀廣播相容 (預設值為 0)。

  • maxval (RealArray) – 選填，範圍的最大值 (排除)，與形狀廣播相容 (預設值為 1)。

傳回:

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

傳回型別:

Array