jax.ref.new_ref#

jax.ref.new_ref(init_val,*,memory_space=None)[source]#

Create a mutable array reference with initial valueinit_val.

For more discussion, see theRef guide.

Parameters:

• init_val (Any) – Ajax.Array representing the initial stateof the buffer.

• memory_space (Any) – An optional memory space attribute for the Ref.

Returns:

Ajax.ref.Ref containing a reference to a mutable buffer.

Return type:

Ref