jax.extend.core.Var

class jax.extend.core.Var(suffix, aval)

Parameters:

• suffix (str)

• aval (AbstractValue)

__init__(suffix, aval)

Parameters:

• suffix (str)

• aval (AbstractValue)

Methods

__init__(suffix, aval)
pretty_print(context, *[, print_dtype])

Attributes

count
suffix
aval