jax.extend.linear_util.transformation# jax.extend.linear_util.transformation = functools.partial(<class 'functools.partial'>, <function transformation>)[源代码]# 参数： fun (WrappedFun) 返回类型： WrappedFun
