jax.extend.core.Jaxpr

类 jax.extend.core.Jaxpr(constvars, invars, outvars, eqns, effects=frozenset({}), debug_info=None, is_high=False)

参数:

• constvars (序列[Var])

• invars (序列[Var])

• outvars (序列[Atom])

• eqns (序列[JaxprEqn])

• effects (Effects)

• debug_info (DebugInfo)

• is_high (bool)

__init__(constvars, invars, outvars, eqns, effects=frozenset({}), debug_info=None, is_high=False)

参数:

• constvars (序列[Var]) – 为常量引入的变量列表。数组常量会被此类变量替换，而标量常量则保持内联。

• invars (序列[Var]) – 输入变量列表。结合 constvars 和 invars，它们是 Jaxpr 的输入。

• outvars (序列[Atom]) – 输出原子列表。

• eqns (序列[JaxprEqn]) – 方程列表。

• effects (Effects) – 效果集。jaxpr 上的效果是每个方程效果的并集的超集。

• debug_info (DebugInfo) – 调试信息。

• is_high (bool)

方法

__init__(constvars, invars, outvars, eqns[, ...])
pretty_print(*[, source_info, print_shapes, ...])
replace(**kwargs)

属性

constvars
debug_info
effects
eqns
invars
is_high
outvars