jax.ref.new_ref

jax.ref.new_ref(init_val, *, memory_space=None)

创建一个具有初始值 init_val 的可变数组引用。

有关更多讨论，请参阅 Ref 指南。

参数:

• init_val – 一个 jax.Array，表示缓冲区的初始状态。

• memory_space (Any) – Ref 的可选内存空间属性。

返回:

一个包含可变缓冲区引用的 jax.ref.Ref。

返回类型:

Ref