Skip to content

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 只能使用一次。



回到顶部