jax.ref.ArrayRef#

class jax.ref.ArrayRef(aval, buf)[source]#

Mutable array reference.

In most cases this should not be constructed directly, but rather via jax.ref.array_ref(). For examples of how this can be used, refer to the ArrayRef guide.

__init__(aval, buf)[source]#

Methods

__init__(aval, buf)

addupdate(x[, idx])

unsafe_buffer_pointer()

Attributes

at

aval

committed

dtype

format

shape

sharding