torch.utils.dlpack

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

解码DLPack到张量。

Parameters
    **dlpack** - 与dltensor一个PyCapsule对象

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

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

返回表示张量DLPack。

Parameters

    **tensor** - 要导出的张量

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


Copyright © ibooker.org.cn 2019 all right reserved,由 ApacheCN 团队提供支持该文件修订时间: 2019-10-08 15:10:38

results matching ""

    No results matching ""

    results matching ""

      No results matching ""