torch.utils.dlpack

原文: https://pytorch.org/docs/stable/dlpack.html


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

将 DLPack 解码为张量。

参数

dlpack –具有 dltensor 的 PyCapsule 对象

张量将与 dlpack 中表示的对象共享内存。 请注意,每个 dlpack 只能使用一次。


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

返回表示张量的 DLPack。

Parameters

张量 –要导出的张量

dlpack 共享张量内存。 请注意,每个 dlpack 只能使用一次。


Copyright © ibooker.org.cn 2019 all right reserved,由 ApacheCN 团队提供支持该文件修订时间: 2020-10-22 03:30:45

results matching ""

    No results matching ""

    results matching ""

      No results matching ""