jax.ref.Ref

class jax.ref.Ref(aval, refs)

可变数组引用。

在大多数情况下，不应直接构造此对象，而应通过 jax.ref.new_ref() 进行构造。有关如何使用此对象，请参阅 Ref 指南。

__init__(aval, refs)

方法

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

属性

at
aval
committed
dtype
format
shape
sharding