PyTorch 1.0 中文文档:torch.utils.dlpack

2022-05-07 14:22:58 浏览数 (1)

译者:kunwuz

代码语言:javascript复制
torch.utils.dlpack.from_dlpack(dlpack) → Tensor

将DLPack解码成Tensor张量。

参数:

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

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

代码语言:javascript复制
torch.utils.dlpack.to_dlpack(tensor) → PyCapsule

返回一个表示张量的DLPack。

参数:

tensor –一个用来输出的tensor张量

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

阅读全文/改进本文

0 人点赞