jax.ffi.register_ffi_type_id

jax.ffi.register_ffi_type_id(name, obj, platform='cpu')

为FFI目标注册自定义类型ID。

参数:

• name (str) – 类型ID的名称。 此名称在进程中必须是唯一的。

• obj (Any) – 封装指向类型ID的指针的 PyCapsule 对象。

• platform (str) – 目标平台。

返回类型:

无