from_dlpack

dragon.vm.torch.utils.dlpack.from_dlpack(dlpack)[source]

Create a tensor sharing the dlpack data.

Parameters:
  • dlpack (PyCapsule) – The capsule object of a dlpack tensor.
Returns:

dragon.vm.torch.Tensor – The tensor with the dlpack data.