jax.lax.rng_比特生成器

jax.lax.rng_比特生成器#

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

无状态的伪随机数生成器。实验性质，不建议使用。

返回使用平台特定默认算法或指定算法生成的具有指定形状和数据类型的均匀分布的随机位（要求为整数类型）。

它提供了直接访问 XLA 暴露的 RngBitGenerator 原语（https://www.tensorflow.org/xla/operation_semantics#rngbitgenerator），用于低级 API 访问。

大多数用户应使用 jax.random 以获得更稳定且用户友好的 API。

目录