torch.utils.dlpack¶
-
torch.utils.dlpack.
from_dlpack
(ext_tensor) → 张量[来源]¶ 通过协议将 Tensor 从外部库转换为 a 。
torch.Tensor
__dlpack__
张量将与所表示的对象共享内存 在 DLPack 中。
警告
每个胶囊只叫 from_dlpack 一次。使用时的行为 在同一个胶囊上多次是未定义的。
- 参数
ext_tensor (object with the object with __dlpack__ attribute or DLPack capsule) - 要转换的张量或 DLPack 胶囊。
-
torch.utils.dlpack.
to_dlpack
(tensor) → PyCapsule¶ 返回表示张量的 DLPack。
- 参数
tensor – 要导出的张量
DLPack 共享张量内存。 请注意,每个 DLPack 只能使用一次。