jax.extend.core.jaxpr_as_fun#

jax.extend.core.jaxpr_as_fun = functools.partial(<class 'functools.partial'>, <function jaxpr_as_fun>)[原始碼]#
參數:

closed_jaxpr (ClosedJaxpr)