torch.utils.dlpack

原文： https://pytorch.org/docs/stable/dlpack.html

torch.utils.dlpack.from_dlpack(dlpack) → Tensor¶

将 DLPack 解码为张量。

参数

dlpack –具有 dltensor 的 PyCapsule 对象

张量将与 dlpack 中表示的对象共享内存。请注意，每个 dlpack 只能使用一次。

torch.utils.dlpack.to_dlpack(tensor) → PyCapsule¶

返回表示张量的 DLPack。

Parameters

张量 –要导出的张量

dlpack 共享张量内存。请注意，每个 dlpack 只能使用一次。