jax.ref.AbstractRef#

class jax.ref.AbstractRef(inner_aval, memory_space=None)[源代码]#

抽象的可变数组引用。

有关更多信息，请参阅 Ref 指南。

参数:

  • inner_aval (core.AbstractValue)

  • memory_space (Any)

__init__(inner_aval, memory_space=None)[源代码]#
参数:

  • inner_aval (core.AbstractValue)

  • memory_space (Any)

方法

__init__(inner_aval[, memory_space])

at_least_vspace()

lo_ty()

lo_ty_qdd(qdd)

lower_val(ref)

normalize()

raise_val(*vals)

str_short([short_dtypes, mesh_axis_types])

strip_weak_type()

to_tangent_aval()

update([inner_aval, memory_space])

update_vma(vma)

update_weak_type(weak_type)

属性

inner_aval

memory_space

T

addupdate

at

bitcast

dtype

get

has_qdd

is_high

bool(x) -> bool

ndim

reshape

set

shape

sharding

size

swap

transpose

vma

weak_type