jax.core.ClosedJaxpr# class jax.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 in_avals jaxpr literals out_avals
