jax.random.PRNGKey#

jax.random.PRNGKey(seed, *, impl=None)[source]#

使用整數種子建立舊版 PRNG 金鑰。

此函數產生舊式舊版 PRNG 金鑰，它們是 dtype 為 uint32 的陣列。如需更多資訊，請參閱 PRNG 金鑰 章節中的註解。在可能的情況下，建議改用 jax.random.key()。

產生的金鑰不帶有 PRNG 實作。傳回的金鑰符合選用的 impl 引數給定的實作，否則由 jax_default_prng_impl 設定旗標決定。呼叫者必須確保在將此金鑰作為引數傳遞給其他函數 (例如 jax.random.split 和 jax.random.normal) 時，設定相同的實作作為預設值。

參數:

  • seed (int | ArrayLike) – 用作金鑰值的 64 位元或 32 位元整數。

  • impl (PRNGSpecDesc | None | None) – 選用字串，指定 PRNG 實作 (例如 'threefry2x32')

回傳:

一個 PRNG 金鑰，可供隨機函數以及 split 和 fold_in 使用。

回傳型別:

陣列