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 只能使用一次。 我们一直在努力 apachecn/AiLearning 为正常使用来必力评论功能请激活JavaScript