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.