jax.ref.Ref

class jax.ref.Ref(aval, refs)

Mutable array reference.

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

__init__(aval, refs)

Methods

__init__(aval, refs)
addupdate(x[, idx])
unsafe_buffer_pointer()

Attributes

at
aval
committed
dtype
format
ndim
shape
sharding