to_dlpack¶

dragon.vm.torch.utils.dlpack.to_dlpack(tensor)[source]¶

Return a dlpack tensor sharing the data.

Parameters:

• tensor (dragon.vm.torch.Tensor) – The tensor to provide data.

Returns:

PyCapsule – The dlpack tensor object.