jax.ref 模块

jax.ref 提供了与 ArrayRef 交互的 API。

API

AbstractRef(inner_aval[, memory_space])	抽象可变数组引用。
Ref(aval, refs)	可变数组引用。
freeze(ref)	使给定引用失效并返回其最终值。
get(ref[, idx])	从 Ref 中读取值。
new_ref(init_val, *[, memory_space])	使用初始值 init_val 创建一个可变数组引用。
set(ref, idx, value)	就地设置 Ref 中的值。
swap(ref, idx, value[, _function_name])	就地更新数组值，同时返回先前的值。
addupdate(ref, idx, x)	就地向 Ref 中的元素添加值。