Movatterモバイル変換


[0]ホーム

URL:


Skip to main content
Ctrl+K
JAX  documentation - Home

jax.extend.linear_util module

jax.extend.linear_util module#

StoreException

WrappedFun(f, f_transformed, transforms, ...)

Represents a functionf to whichtransforms are to be applied.

cache(call, *[, explain])

Memoization decorator for functions taking a WrappedFun as first argument.

merge_linear_aux(aux1, aux2)

transformation

transformation_with_aux

wrap_init(f[, params, debug_info])


[8]ページ先頭

©2009-2025 Movatter.jp