jax.random.bits

jax.random.bits(key, shape=(), dtype=None)

以無號整數形式取樣均勻位元。

參數:

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

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

• dtype (DTypeLikeUInt | None | None) – 選用，用於傳回值的無號整數 dtype (如果 jax_enable_x64 為 true，則預設為 uint64，否則為 uint32)。

傳回:

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

傳回型別:

Array