jax.extend.linear_util 模块

`jax.extend.linear_util` 模块#

`可调用对象`()
`StoreException`
`WrappedFun`(f, f_transformed, transforms, ...)	表示要应用 transforms 的函数 f。
`cache`(call, *[, explain])	用于以 WrappedFun 作为第一个参数的函数的记忆化装饰器。
`merge_linear_aux`(aux1, aux2)
`transformation`
`transformation2`	为 WrappedFun 添加一个额外的转换。
`transformation_with_aux`
`transformation_with_aux2`	为 WrappedFun 添加一个带有辅助输出的额外转换。
`wrap_init`(f[, params, debug_info])