jax.extend.core.ClosedJaxpr# class jax.extend.core.ClosedJaxpr(jaxpr, consts)[source]# 參數: jaxpr (Jaxpr) consts (Sequence) __init__(jaxpr, consts)[source]# 參數: jaxpr (Jaxpr) consts (Sequence) 方法 __init__(jaxpr, consts) map_jaxpr(f) pretty_print(*[, source_info, print_shapes, ...]) replace(*[, jaxpr, consts]) 屬性 consts effects eqns in_avals jaxpr literals out_avals