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, *, out_sharding=None)[source]#

无状态 PRNG 比特生成器。实验性的，不鼓励使用。

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

它提供对 XLA 公开的 RngBitGenerator 原始类型的直接访问 (https://www.openxla.org/xla/operation_semantics#rngbitgenerator) 用于低级别 API 访问。

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