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)[源代码]#

无状态 PRNG 位生成器。实验性的，不建议使用。

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

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

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