jax.random.wrap_key_data

目錄

jax.random.wrap_key_data#

jax.random.wrap_key_data(key_bits_array, *, impl=None)[原始碼]#

將金鑰資料位元陣列包裝成 PRNG 金鑰陣列。

參數:

key_bits_array (Array) – 一個 uint32 陣列，其尾隨形狀對應於 impl 指定的 PRNG 實作的金鑰形狀。
impl (PRNGSpecDesc | None | None) – 選項，指定 PRNG 實作，如 random.key 中所示。

回傳:

一個 PRNG 金鑰陣列，其 dtype 是 jax.dtypes.prng_key 的子型別: 對應於 impl，且其形狀等於 key_bits_array.shape 的前導形狀，直到金鑰位元維度。