当前位置：首页 > 文档资料 > PyTorch 1.0 中文文档 & 教程 >

torch.utils.dlpack

优质

小牛编辑

132浏览

2023-12-01

译者：kunwuz

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

将DLPack解码成Tensor张量。

参数:	dlpack – 一个有着dltensor张量的PyCapsule对象

这个张量会与dlpack对象共享存储空间。注意每个dlpack对象只能使用一次。

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

返回一个表示张量的DLPack。

参数:	tensor –一个用来输出的tensor张量

这个张量会与dlpack对象共享存储空间。注意每个dlpack对象只能使用一次。

免责声明：以上内容版权归原作者所有，如有侵犯您的原创版权请告知，我们将尽快删除相关内容。感谢每一位辛勤著写的作者，感谢每一位的分享。