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（需要为整数类型）的均匀分布的随机位。

它为低级 API 访问提供了 XLA 公开的 RngBitGenerator primitive 的直接访问 (https://tensorflowcn.cn/xla/operation_semantics#rngbitgenerator)。

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