jax.extend.linear_util.wrap_init# jax.extend.linear_util.wrap_init(f, params=None)[source]# 将函数 f 包装为 WrappedFun，适合进行变换。 返回类型: WrappedFun
