jax.extend.core.ClosedJaxpr

类 jax.extend.core.ClosedJaxpr(jaxpr, consts)

参数:

• jaxpr (Jaxpr)

• consts (Sequence)

__init__(jaxpr, consts)

参数:

• jaxpr (Jaxpr)

• consts (Sequence)

方法

__init__(jaxpr, consts)
map_jaxpr(f)
pretty_print(*[, source_info, print_shapes, ...])
replace(*[, jaxpr, consts])

属性

consts
effects
eqns
final_aval_qdds
in_aval_qdds
in_avals
jaxpr
literals
out_avals