jax.core.ClosedJaxpr# class jax.core.ClosedJaxpr(jaxpr, consts)[source]# 参数： : jaxpr (Jaxpr) consts (序列) __init__(jaxpr, consts)[source]# 参数： : jaxpr (Jaxpr) consts (序列) 方法 __init__(jaxpr, consts) map_jaxpr(f) pretty_print(*[, source_info, print_shapes, ...]) replace(*[, jaxpr, consts]) 属性 consts effects eqns in_avals jaxpr literals out_avals
