jax.dlpack 模块

from_dlpack(external_array[, device, copy])	返回 DLPack 张量的 Array 表示。
to_dlpack(x[, stream, src_device, ...])	返回一个 DLPack 张量，它封装了一个 Array x。