dlpack.md 628 字节
Newer Older
W
wizardforcel 已提交
1 2
# torch.utils.dlpack

飞龙 已提交
3 4
> 译者:[kunwuz](https://github.com/kunwuz)

W
wizardforcel 已提交
5
```py
W
wizardforcel 已提交
6
torch.utils.dlpack.from_dlpack(dlpack)  Tensor
W
wizardforcel 已提交
7 8
```

K
kunwuz  
kunwu 已提交
9
将DLPack解码成Tensor张量。
W
wizardforcel 已提交
10

K
kunwuz  
kunwu 已提交
11
| 参数: | **dlpack** – 一个有着dltensor张量的PyCapsule对象 |
W
wizardforcel 已提交
12 13
| --- | --- |

K
kunwuz  
kunwu 已提交
14
这个张量会与dlpack对象共享存储空间。注意每个dlpack对象只能使用一次。
W
wizardforcel 已提交
15 16

```py
W
wizardforcel 已提交
17
torch.utils.dlpack.to_dlpack(tensor)  PyCapsule
W
wizardforcel 已提交
18 19
```

K
kunwuz  
kunwu 已提交
20
返回一个表示张量的DLPack。
W
wizardforcel 已提交
21

K
kunwu 已提交
22
| 参数: | **tensor** –一个用来输出的tensor张量 |
W
wizardforcel 已提交
23 24
| --- | --- |

K
kunwuz  
kunwu 已提交
25
这个张量会与dlpack对象共享存储空间。注意每个dlpack对象只能使用一次。
W
wizardforcel 已提交
26