jax.extend.linear_util.wrap_init# jax.extend.linear_util.wrap_init(f, params=None)[源代码][源代码]# 将函数 f 包装为 WrappedFun，适用于转换。 返回类型: WrappedFun
