jax.ffi.register_ffi_type_id#

jax.ffi.register_ffi_type_id(name, obj, platform='cpu')[source]#

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

參數:

  • name (str) – 類型 ID 的名稱。此名稱在進程中必須是唯一的。

  • obj (Any) – 一個封裝指向類型 ID 指標的 PyCapsule 物件。

  • platform (str) – 目標平台。

返回類型:

無