jax.extend.linear_util.wrap_init

jax.extend.linear_util.wrap_init(f, params=None, *, debug_info=None)

参数:

f (Callable)

返回类型:

WrappedFun