目录

torch.utils.dlpack

torch.utils.dlpack.from_dlpack(ext_tensor张量[来源]

通过协议将 Tensor 从外部库转换为 a 。torch.Tensor__dlpack__

张量将与所表示的对象共享内存 在 DLPack 中。

警告

每个胶囊只叫 from_dlpack 一次。使用时的行为 在同一个胶囊上多次是未定义的。

参数

ext_tensorobject with the object with __dlpack__ attribute or DLPack capsule) - 要转换的张量或 DLPack 胶囊。

torch.utils.dlpack.to_dlpack(tensorPyCapsule

返回表示张量的 DLPack。

参数

tensor – 要导出的张量

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

文档

访问 PyTorch 的全面开发人员文档

查看文档

教程

获取面向初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并解答您的问题

查看资源