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 位生成器。实验性，不建议使用。

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

它提供了对 XLA (https://tensorflowcn.cn/xla/operation_semantics#rngbitgenerator) 公开的 RngBitGenerator 原语的直接访问，用于低级 API 访问。

大多数用户应该使用 jax.random，因为它提供了稳定的、更友好的 API。

内容