jax.extend.linear_util.transformation_with_aux

jax.extend.linear_util.transformation_with_aux = functools.partial(<class 'functools.partial'>, <function transformation_with_aux>)

參數:

fun (WrappedFun)

回傳型別:

WrappedFun