jax.random.split

jax.random.split(key, num=2)

透過新增前導軸，將 PRNG 金鑰分割成 num 個新的金鑰。

參數：

• key (ArrayLike) – PRNG 金鑰 (來自 key、split、fold_in)。

• num (int | tuple[int, ...]) – 選填，正整數（或整數元組），表示要產生的金鑰數量（或形狀）。預設值為 2。

回傳：

包含 num 個新 PRNG 金鑰的類陣列物件。

回傳類型：

Array