jax.extend.core.Jaxpr#

class jax.extend.core.Jaxpr(constvars, invars, outvars, eqns, effects=frozenset({}), debug_info=None)[原始碼]#
參數:

  • constvars (Sequence[Var])

  • invars (Sequence[Var])

  • outvars (Sequence[Atom])

  • eqns (Sequence[JaxprEqn])

  • effects (Effects)

  • debug_info (JaxprDebugInfo | None)

__init__(constvars, invars, outvars, eqns, effects=frozenset({}), debug_info=None)[原始碼]#
參數:

  • constvars (Sequence[Var]) – 為常數引入的變數列表。陣列常數會被替換為此類變數，而純量常數則保持內聯。

  • invars (Sequence[Var]) – 輸入變數列表。constvars 和 invars 一起構成 Jaxpr 的輸入。

  • outvars (Sequence[Atom]) – 輸出原子列表。

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

  • effects (Effects) – 效果集。jaxpr 上的效果是每個方程式效果的聯集的超集。

  • debug_info (JaxprDebugInfo | None | None) – 選用的 JaxprDebugInfo。

方法

__init__(constvars, invars, outvars, eqns[, ...])

pretty_print(*[, source_info, print_shapes, ...])

replace(**kwargs)

屬性

constvars

debug_info

effects

eqns

invars

outvars