- 博客(3)
- 资源 (3)
- 收藏
- 关注
原创 Haiku 库中的get_state和set_state及get_parameter函数
haiku.get_state` 和 `haiku.set_state` 是 DeepMind 的 Haiku 库中的两个核心函数,用于在 JAX 框架中管理可变状态。- 功能: `haiku.set_state` 用于更新模型的状态。在这个示例中,每次调用模型时,状态 `counter` 都会增加1,反映出在前向传播过程中状态的变化。- `value`: 要设置的新状态值,应该与 `get_state` 获取的状态形状和类型匹配。- `dtype`: 状态的数据类型,如 `jnp.float32`。
2024-08-22 13:12:29
91
原创 jax.random.PRNGKey用法
jax.random.PRNGKey` 是 JAX 管理随机数生成的核心工具,通过合理的 key 管理和拆分,可以在复杂的计算中确保随机数生成的可控性和可重复性。- `random.split(key)` 将原始 key 拆分成两个新的 PRNG key,`key1` 和 `key2`,它们可以独立用于随机数生成。- JAX 的 PRNG key 是不可变的,在生成随机数时,务必注意每次都拆分 key 以生成新的 key,这样可以保证每次生成的随机数都是独立的。这个 key 保证了随机数生成的可重复性。
2024-08-19 17:06:41
532
原创 numpy模块中的searchsorted函数
用于在一个已排序的一维数组中查找指定值的插入位置,以确保插入后数组仍然有序。或。它适用于单个值或多个值的查找,并返回插入位置的索引。
2024-08-18 11:52:24
232
1
点阵字模生成原理与方法+点阵取模软件
2012-05-05
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人