jax.ffi 模組#

ffi_call()

呼叫外部函數介面 (FFI) 目標。

ffi_lowering(call_target_name, *[, ...])

為外部函數介面 (FFI) 目標建立降低規則。

pycapsule(funcptr)

將 ctypes 函數指標包裝在 PyCapsule 中。

register_ffi_target(name, fn[, platform, ...])

註冊外部函數目標。

register_ffi_type_id(name, obj[, platform])

為 FFI 目標註冊自訂型別 ID。

jax.extend.ffi 模組 (已棄用)#

jax.extend.ffi 模組已移至 jax.ffi,應該改用該匯入路徑,但在舊版匯入被棄用期間,這些函數仍在此處記錄。

ffi_call()

呼叫外部函數介面 (FFI) 目標。

ffi_lowering(call_target_name, *[, ...])

為外部函數介面 (FFI) 目標建立降低規則。

pycapsule(funcptr)

將 ctypes 函數指標包裝在 PyCapsule 中。

register_ffi_target(name, fn[, platform, ...])

註冊外部函數目標。