gpt4 book ai didi

jax - 如何在jax中生成0到1之间的随机数?

转载 作者:行者123 更新时间:2023-12-05 04:29:31 28 4
gpt4 key购买 nike

如何在 jax 中生成 0 到 1 之间的随机数?基本上,我希望从 jax 中的 numpy 复制以下函数。

np.random.random(1000)

最佳答案

jax 中的等价物是

from jax import random
key = random.PRNGKey(758493) # Random seed is explicit in JAX
random.uniform(key, shape=(1000,))

有关详细信息,请参阅 jax.random module 的文档.

另请注意,JAX 的随机数生成器不维护任何类型的全局状态,因此您需要以不同于您在 NumPy 中习惯的方式来考虑它。有关这方面的更多背景信息,请参阅 JAX Sharp Bits: Random Numbers .

关于jax - 如何在jax中生成0到1之间的随机数?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/72320999/

28 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com