jax.extend.linear_util.transformation#

jax.extend.linear_util.transformation = functools.partial(<class 'functools.partial'>, <function transformation>)[原始碼]#
參數:

fun (WrappedFun)

回傳類型:

WrappedFun