译者: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张量
阅读全文/改进本文
本文分享自 作者个人站点/博客 前往查看
如有侵权,请联系 cloudcommunity@tencent.com 删除。
本文参与 腾讯云自媒体同步曝光计划 ,欢迎热爱写作的你一起参与!