jax.lax.rng_bit_generator

jax.lax.rng_bit_generator#

jax.lax.rng_bit_generator(key, shape, dtype=<class 'numpy.uint32'>, algorithm=RandomAlgorithm.RNG_DEFAULT)[source]#

無狀態 PRNG 位元產生器。實驗性質，不建議使用。

使用平台特定的預設演算法或指定的演算法，傳回具有指定形狀和 dtype（需要為整數型別）的均勻分佈隨機位元。

它提供對 XLA 公開的 RngBitGenerator 基本運算 (https://tensorflow.dev.org.tw/xla/operation_semantics#rngbitgenerator) 的直接存取，以進行低階 API 存取。

大多數使用者應改用 jax.random，以獲得更穩定且更使用者友善的 API。