jax.dlpack 模組

from_dlpack(external_array[, device, copy])	傳回 DLPack 張量的 Array 表示法。
to_dlpack(x[, stream, src_device, ...])	傳回封裝 Array x 的 DLPack 張量。