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