torch.utils.dlpack.from_dlpack(dlpack) → Tensor Decodes a DLPack to a tensor.
dlpack – a PyCapsule object with the dltensor
The tensor will share the memory with the object represented in the dlpack. Note that each dlpack can only be consumed once.
torch.utils.dlpack.to_dlpack(tensor) → PyCapsule Returns a DLPack representing the tensor.
tensor – a tensor to be exported
The dlpack shares the tensors memory. Note that each dlpack can only be consumed once.
© 2019 Torch Contributors
Licensed under the 3-clause BSD License.
https://pytorch.org/docs/1.7.0/dlpack.html