jax.random.clone

jax.random.clone(key)

克隆一个密钥以供重复使用

在密钥重复使用检查的上下文中（请参见 jax.experimental.key_reuse），此函数充当标识。

示例

>>> import jax
>>> key = jax.random.key(0)
>>> data = jax.random.uniform(key)
>>> cloned_key = jax.random.clone(key)
>>> same_data = jax.random.uniform(cloned_key)
>>> assert data == same_data